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)
-                image = self.resize_image(sample_image_path, new_width, new_height)
+                image = self.resize_image(sample_image_path, 100, 100)
                 st.image(image)
                 if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx}'):
                     self.process_new_image(sample_image_path, image, kbvqa)
@@ -71,6 +71,27 @@ class InferenceRunner(StateManager):
 
     def run_inference(self):
         self.set_up_widgets()
+
+
+
+        
+        self.col1.write("Choose from sample images:")
+        cols = self.col1.columns(len(self.sample_images))
+        for idx, sample_image_path in enumerate(self.sample_images):
+            with cols[idx]:
+                image = Image.open(sample_image_path)
+                image = self.resize_image(sample_image_path, 100, 100)
+                st.image(image)
+                if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx}'):
+                    self.process_new_image(sample_image_path, image, kbvqa)
+
+
+
+
+
+
+
+        
         st.session_state['settings_changed'] = self.has_state_changed()
         if st.session_state['settings_changed']:
             self.col1.warning("Model settings have changed, please reload the model, this will take a second .. ")