Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -96,21 +96,29 @@ def process_new_image(image_key, image, kbvqa):
 def run_inference():
     st.title("Run Inference")
 
-    method = st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0)
-    detection_model = st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1)
+    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 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 = st.slider("Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=default_confidence, step=0.1)
+    state_manager.set_slider_value(text="Select minimum detection confidence level", 
+                                                    min_value=0.1, 
+                                                    max_value=0.9, 
+                                                    value=default_confidence, 
+                                                    step=0.1, 
+                                                    slider_key_name='confidence_level',
+                                                    )
 
-    state_manager.update_model_settings(detection_model=detection_model, confidence_level=confidence_level, selected_method=method)
+   # 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 need_model_reload else "Load Model"
+   # need_model_reload = state_manager.check_settings_changed(method, detection_model, confidence_level) and state_manager.is_model_loaded()
+    need_model_reload = 
+    button_label = "Reload Model" if st.session_state['kbvqa'] else "Load Model"
 
     if method == "Fine-Tuned Model":
         if st.button(button_label):
-            if not need_model_reload:
-                st write(st.session_state['kbvqa'])
+            if button_label == "Load Model" and :
+               # st write(st.session_state['kbvqa'])
                 st.write("Model already loaded.")
             else:
                 st.text("Loading the model, please wait...")