Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -36,7 +36,7 @@ class InferenceRunner(StateManager):
             with cols[idx]:
                 image = Image.open(sample_image_path)
                 image_for_display = copy.deepcopy(image) # resize just for the display control without changing original image
-                image_for_display = self.resize_image(image_for_display, 500, 500)
+                image_for_display = self.resize_image(image_for_display, 300, 300)
                 st.image(image_for_display)
                 if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx}'):
                     self.process_new_image(sample_image_path, image, kbvqa)
@@ -49,7 +49,7 @@ class InferenceRunner(StateManager):
         # Display and interact with each uploaded/selected image
         for image_key, image_data in self.get_images_data().items():
             image_for_display = copy.deepcopy(image_data['image'])  # resize just for the display control without changing original image
-            image_for_display = self.resize_image(image_for_display, 800, 800)
+            image_for_display = self.resize_image(image_for_display, 500, 500)
             self.col2.image(image_for_display, caption=f'Uploaded Image: {image_key[-11:]}')
             if not image_data['analysis_done']:
                 self.col2.text("Cool image, please click 'Analyze Image'..")