Update my_model/state_manager.py

my_model/state_manager.py

@@ -31,7 +31,7 @@ class StateManager:
         if 'force_reload_button_clicked' not in st.session_state:
             st.session_state['force_reload_button_clicked'] = False
         if 'time_taken_to_load_model' not in st.session_state:
-            st.session_state['time_taken_to_load_model'] = 0
+            st.session_state['time_taken_to_load_model'] = None
         if "settings_changed" not in st.session_state:
             st.session_state['settings_changed'] = self.settings_changed
         if 'model_loaded' not in st.session_state:
@@ -291,8 +291,10 @@ class StateManager:
         """
         img = copy.deepcopy(image)
         st.text("Analyzing the image .. ")
-        caption = st.session_state.kbvqa.get_caption(img)
-        image_with_boxes, detected_objects_str = self.session_state.kbvqa.detect_objects(img)
+        caption = st.session_state['kbvqa'].get_caption(img)
+        st.text("DDDDDDDDD")
+        image_with_boxes, detected_objects_str = self.session_state['kbvqa'].detect_objects(img)
+        st.text("TTTTTT")
         return caption, detected_objects_str, image_with_boxes