Update my_model/state_manager.py

my_model/state_manager.py

@@ -196,7 +196,7 @@ 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
+        return 'kbvqa' in st.session_state and st.session_state['kbvqa'] is not None and st.session_state.kbvqa.all_models_loaded
 
     
     def reload_detection_model(self):