Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -102,7 +102,7 @@ def run_inference():
 
     method = st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
     detection_model = st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
-    default_confidence = 0.2 if st.session_state['detection_model'] == "yolov5" else 0.4
+    default_confidence = 0.2 if detection_model == "yolov5" else 0.4
    # confidence_level = st.slider("Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=default_confidence, step=0.1)
     confidence_level = state_manager.set_slider_value(text="Select minimum detection confidence level", 
                                                     min_value=0.1,