Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -58,7 +58,7 @@ class StateManager:
             #self.update_model_settings(detection_model, 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}
-            
+            st.session_state['kbvqa'] = "Reload Model"
             free_gpu_resources()
         except Exception as e:
             st.error(f"Error loading model: {e}")