Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -21,16 +21,8 @@ class StateManager:
         if 'confidence_level' not in st.session_state:
             st.session_state['confidence_level'] = None
 
-        st.text("session state iniializaed")
 
 
-    def update_model_settings(self, detection_model=None, confidence_level=None, selected_method=None):
-        if detection_model is not None:
-            st.session_state['model_settings']['detection_model'] = detection_model
-        if confidence_level is not None:
-            st.session_state['model_settings']['confidence_level'] = confidence_level
-        if selected_method is not None:
-            st.session_state['model_settings']['selected_method'] = selected_method
 
     def set_slider_value(self, text, min_value, max_value, value, step, slider_key_name):
 
@@ -79,7 +71,7 @@ class StateManager:
             if self.is_model_loaded():
                 prepare_kbvqa_model(detection_model, only_reload_detection_model=True)
                 st.session_state['kbvqa'].detection_confidence = confidence_level
-                self.update_model_settings(detection_model, confidence_level)
+                #self.update_model_settings(detection_model, confidence_level)
             free_gpu_resources()
         except Exception as e:
             st.error(f"Error reloading detection model: {e}")