Update my_model/state_manager.py

my_model/state_manager.py

@@ -18,6 +18,12 @@ class StateManager:
     def initialize_state(self):
         if "previous_state" not in st.session_state:
             st.session_state['previous_state'] = {'method': None, 'detection_model': None, 'confidence_level': None}
+        if 'method' not in st.session_state:
+            st.session_state['method'] = None
+        if 'method' not in st.session_state:
+            st.session_state['confidence_level'] = None
+        if 'method' not in st.session_state:
+            st.session_state['detection_model'] = None
         if 'images_data' not in st.session_state:
             st.session_state['images_data'] = {}
         if 'kbvqa' not in st.session_state:
@@ -40,6 +46,7 @@ class StateManager:
             st.session_state['model_loaded'] = self.is_model_loaded
 
 
+