Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -82,9 +82,11 @@ class InferenceRunner(StateManager):
                 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}', size='sm'):
+                if st.button(st.image(image), key=f'sample_{idx}'):
                     self.process_new_image(sample_image_path, image, kbvqa)
 
+                    
+