Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -8,12 +8,13 @@ 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:
             st.session_state['kbvqa'] = None
-        if "button_lablel" not in st.session_state: 
-            st.session_state['button_lablel'] = "Load Model"
+        if "button_label" not in st.session_state: 
+            st.session_state['button_label'] = "Load Model"
         if "previous_state" not in st.session_state:
             st.session_state['previous_state'] = {}
             
@@ -58,7 +59,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['button_lablel'] = "Reload Model"
+            st.session_state['button_label'] = "Reload Model"
             free_gpu_resources()
         except Exception as e:
             st.error(f"Error loading model: {e}")