Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -59,7 +59,7 @@ class InferenceRunner(StateManager):
 
                                 caption, detected_objects_str, image_with_boxes = self.analyze_image(image_data['image'], kbvqa)
                                 self.update_image_data(image_key, caption, detected_objects_str, True)
-                                st.session_state['loading_in_progress'] = False
+                            st.session_state['loading_in_progress'] = False
 
         
                     # Initialize qa_history for each image
@@ -151,5 +151,6 @@ class InferenceRunner(StateManager):
             st.session_state['loading_in_progress'] = False
             free_gpu_resources()
             self.image_qa_app(self.get_model())
+            st.write(st.session_state['loading_in_progress'])