Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -55,8 +55,8 @@ class StateManager:
         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.state_session.detection_model)
-            st.session_state['kbvqa'].detection_confidence = st.state_session.confidence_level
+            st.session_state['kbvqa'] = prepare_kbvqa_model(st.session_state.detection_model)
+            st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
             #self.update_model_settings(detection_model, confidence_level)
             st.text("Model is ready for inference.")
             free_gpu_resources()