Fix Streamlit Inference model suffix bug (#14621)

Co-authored-by: malopez <miguelangel.lopez@solute.es>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pull/14633/head^2
Miguel Angel Lopez 7 months ago committed by GitHub
parent 03225fce9a
commit 23ce08791f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      ultralytics/solutions/streamlit_inference.py

@ -69,7 +69,7 @@ def inference(model=None):
# Add dropdown menu for model selection
available_models = [x.replace("yolo", "YOLO") for x in GITHUB_ASSETS_STEMS if x.startswith("yolov8")]
if model:
available_models.insert(0, model)
available_models.insert(0, model.split(".pt")[0]) # insert model without suffix as *.pt is added later
selected_model = st.sidebar.selectbox("Model", available_models)
with st.spinner("Model is downloading..."):

Loading…
Cancel
Save