Update my_model/state_manager.py

my_model/state_manager.py

@@ -136,14 +136,10 @@ 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}
@@ -161,11 +157,13 @@ class StateManager:
         free_gpu_resources()
         
         if self.is_model_loaded():
-            del st.session_state['kbvqa']
-            st.write('Model deleted')
-            free_gpu_resources()
-        st.text("AAAAAA")
-        free_gpu_resources()
+            try:
+                del st.session_state['kbvqa']
+                free_gpu_resources()
+            except:
+                free_gpu_resources()
+                pass
+            
 
         
     # Function to check if any session state values have changed