Update my_model/state_manager.py

my_model/state_manager.py

@@ -16,6 +16,8 @@ class StateManager:
         self.col1, self.col2, self.col3 = st.columns([0.2, 0.6, 0.2])  
 
     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 'images_data' not in st.session_state:
             st.session_state['images_data'] = {}
         if 'kbvqa' not in st.session_state:
@@ -36,8 +38,7 @@ class StateManager:
             st.session_state['settings_changed'] = self.settings_changed
         if 'model_loaded' not in st.session_state:
             st.session_state['model_loaded'] = self.is_model_loaded
-        if "previous_state" not in st.session_state:
-            st.session_state['previous_state'] = {'method': None, 'detection_model': None, 'confidence_level': None}
+