Update my_model/state_manager.py

my_model/state_manager.py

@@ -240,7 +240,10 @@ class StateManager:
         Returns:
             bool: True if the model is loaded, False otherwise.
         """
-        return 'kbvqa' in st.session_state and st.session_state['kbvqa'] is not None and st.session_state.kbvqa.all_models_loaded
+        return 'kbvqa' in st.session_state and st.session_state['kbvqa'] is not None and \
+           st.session_state.kbvqa.all_models_loaded \
+           and (st.session_state['previous_state']['method'] is not None 
+                and st.session_state['method'] == st.session_state['previous_state']['method'])
 
     
     def reload_detection_model(self) -> None: