Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -8,7 +8,6 @@ from my_model.KBVQA import KBVQA, prepare_kbvqa_model
 class StateManager:
 
     def initialize_state(self):
-        st.text('states initialized')
         if 'images_data' not in st.session_state:
             st.session_state['images_data'] = {}
         if 'kbvqa' not in st.session_state:
@@ -51,13 +50,14 @@ class StateManager:
         """Load the KBVQA model with specified settings."""
         try:
             free_gpu_resources()
-            st.text("Loading the model, this should take no more than a few minutes, please wait...")
             st.session_state['kbvqa'] = prepare_kbvqa_model()
             st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
             #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['button_label'] = "Reload Model"
+            st.text('button changed')
             free_gpu_resources()
         except Exception as e:
             st.error(f"Error loading model: {e}")