Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -35,7 +35,7 @@ class InferenceRunner(StateManager):
         for idx, sample_image_path in enumerate(self.sample_images):
             with cols[idx]:
                 image = Image.open(sample_image_path)
-                st.image(image, width=50)
+                st.image(image, width=100, height=100)
                 if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx}'):
                     self.process_new_image(sample_image_path, image, kbvqa)
 
@@ -46,7 +46,7 @@ class InferenceRunner(StateManager):
 
         # Display and interact with each uploaded/selected image
         for image_key, image_data in self.get_images_data().items():
-            self.col2.image(image_data['image'], caption=f'Uploaded Image: {image_key[-11:]}', use_column_width=True)
+            self.col2.image(image_data['image'], caption=f'Uploaded Image: {image_key[-11:]}', width=1000)
             if not image_data['analysis_done']:
                 self.col2.text("Cool image, please click 'Analyze Image'..")
                 if self.col2.button('Analyze Image', key=f'analyze_{image_key}'):