Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -98,7 +98,7 @@ def run_inference():
     st.title("Run Inference")
 
 
-    method = st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
+    st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
     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
    # confidence_level = st.slider("Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=default_confidence, step=0.1)
@@ -107,16 +107,17 @@ def run_inference():
                                                     max_value=0.9, 
                                                     value=default_confidence, 
                                                     step=0.1, 
-                                                    slider_key_name='confidence_level',
+                                                    slider_key_name='confidence_level'
                                                     )
 
    # state_manager.update_model_settings(detection_model=detection_model, confidence_level=confidence_level, selected_method=method)
 
 
    # need_model_reload = state_manager.check_settings_changed(method, detection_model, confidence_level) and state_manager.is_model_loaded()
-    
-    button_label = "Reload Model" if st.session_state['kbvqa'] is not None else "Load Model"
     state_manager.display_session_state()
+
+    button_label = "Reload Model" if st.session_state['kbvqa'] is not None else "Load Model"
+
     if st.session_state.method == "Fine-Tuned Model":
         if st.button(button_label):
             if button_label == "Load Model" and state_manager.is_model_loaded():