From 23ce08791f1fa22be93e2d6033f989d1dd13a856 Mon Sep 17 00:00:00 2001 From: Miguel Angel Lopez Date: Tue, 23 Jul 2024 21:57:45 +0200 Subject: [PATCH] Fix Streamlit Inference model suffix bug (#14621) Co-authored-by: malopez Co-authored-by: Glenn Jocher --- ultralytics/solutions/streamlit_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/solutions/streamlit_inference.py b/ultralytics/solutions/streamlit_inference.py index 99916552..85394350 100644 --- a/ultralytics/solutions/streamlit_inference.py +++ b/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..."):