Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -78,6 +78,7 @@ class InferenceRunner(StateManager):
             self.process_new_image(uploaded_image.name, Image.open(uploaded_image), kbvqa)
 
         # Display and interact with each uploaded/selected image
+        self.display_session_state()
         with self.col2:
             for image_key, image_data in self.get_images_data().items():
                 with st.container():
@@ -88,7 +89,7 @@ class InferenceRunner(StateManager):
                         nested_col22.text("Please click 'Analyze Image'..")
                         with nested_col22:
                             if st.button('Analyze Image', key=f'analyze_{image_key}', on_click=self.disable_widgets, disabled=self.is_widget_disabled):
-
+                                
                                 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
@@ -197,5 +198,5 @@ class InferenceRunner(StateManager):
             free_gpu_resources()
             st.session_state['loading_in_progress'] = False
             self.image_qa_app(self.get_model())
-     
+