Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -101,15 +101,15 @@ def run_inference():
     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)
 
-    state_manager.update_model_settings(detection_model, confidence_level, method)
-    settings_changed = state_manager.check_settings_changed(method, detection_model, confidence_level)
+    state_manager.update_model_settings(detection_model=detection_model, confidence_level=confidence_level, selected_method=method)
 
-    need_model_reload = settings_changed and state_manager.is_model_loaded()
+
+    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"
 
     if method == "Fine-Tuned Model":
         if st.button(button_label):
-            if state_manager.is_model_loaded() and not settings_changed:
+            if not need_model_reload:
                 st.write("Model already loaded.")
             else:
                 st.text("Loading the model, please wait...")