Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -78,6 +78,35 @@ class InferenceRunner(StateManager):
         if st.session_state['settings_changed']:
             self.col1.warning("Model settings have changed, please reload the model, this will take a second .. ")
 
+
+
+
+
+
+###################################
+        for idx, sample_image_path in enumerate(self.sample_images):
+            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)
+                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)
+
+        # Image uploader
+        uploaded_image = self.col1.file_uploader("Or upload an Image", type=["png", "jpg", "jpeg"])
+        if uploaded_image is not None:
+            self.process_new_image(uploaded_image.name, Image.open(uploaded_image), kbvqa)
+############################################
+
+
+
+
+
+
+
+            
+
         st.session_state.button_label = "Reload Model" if self.is_model_loaded() and self.settings_changed else "Load Model"
         
         if st.session_state.method == "Fine-Tuned Model":