Update my_model/state_manager.py

my_model/state_manager.py

@@ -136,10 +136,14 @@ class StateManager:
 
     def force_reload_model(self):
         try:
+            st.text("CCCCCC")
             self.delete_model()
+            st.text("BBBBB")
             self.initialize_state()
+            st.text("DDDD")
             free_gpu_resources()
             st.session_state['kbvqa'] = prepare_kbvqa_model(force_reload=True)
+            st.text("EEEEE")
             st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
             # Update the previous state with current session state values
             st.session_state['previous_state'] = {'method': st.session_state.method, 'detection_model': st.session_state.detection_model, 'confidence_level': st.session_state.confidence_level}
@@ -160,6 +164,7 @@ class StateManager:
             del st.session_state['kbvqa']
             st.write('Model deleted')
             free_gpu_resources()
+        st.text("AAAAAA")
         free_gpu_resources()