Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -54,9 +54,10 @@ class InferenceRunner(StateManager):
                     nested_col21.image(image_for_display, caption=f'Uploaded Image: {image_key[-11:]}')
                     if not image_data['analysis_done']:
                         nested_col22.text("Please click 'Analyze Image'..")
-                        if nested_col22.button('Analyze Image', key=f'analyze_{image_key}'):
-                            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)
+                        with nested_col22:
+                            if st.button('Analyze Image', key=f'analyze_{image_key}'):
+                                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)
         
                     # Initialize qa_history for each image
                     qa_history = image_data.get('qa_history', [])
@@ -78,6 +79,7 @@ class InferenceRunner(StateManager):
 
         
     def run_inference(self):
+        
         self.set_up_widgets()
         load_fine_tuned_model = False
         fine_tuned_model_already_loaded = False