Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -12,7 +12,7 @@ class StateManager:
     def initialize_state(self):
         if 'images_data' not in st.session_state:
             st.session_state['images_data'] = {}
-        if 'model_settings' not in st.session_state:
+        if 'method' not in st.session_state:
             st.session_state['method'] = None
         if 'detection_model' not in st.session_state:
             st.session_state['detection_model'] = None