Update my_model/state_manager.py

my_model/state_manager.py

@@ -156,7 +156,7 @@ class StateManager:
         
         free_gpu_resources()
         
-        if st.session_state['kbvqa'] is not None:
+        if self.is_model_loaded():
             del st.session_state['kbvqa']
             st.write('Model deleted')
             free_gpu_resources()