m7mdal7aj commited on
Commit
ab428b9
1 Parent(s): 0d909f9

Update my_model/tabs/run_inference.py

Browse files
Files changed (1) hide show
  1. my_model/tabs/run_inference.py +2 -4
my_model/tabs/run_inference.py CHANGED
@@ -92,13 +92,11 @@ class InferenceRunner(StateManager):
92
  image_for_display = copy.deepcopy(image) # resize just for the display control without changing original image
93
  image_for_display = self.resize_image(image_for_display, 500, 500)
94
  st.image(image_for_display)
95
- if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx}'):
96
- self.process_new_image(sample_image_path, image, kbvqa)
97
 
98
  # Image uploader
99
  uploaded_image = self.col1.file_uploader("Or upload an Image", type=["png", "jpg", "jpeg"])
100
- if uploaded_image is not None:
101
- self.process_new_image(uploaded_image.name, Image.open(uploaded_image), kbvqa)
102
  ############################################
103
 
104
 
 
92
  image_for_display = copy.deepcopy(image) # resize just for the display control without changing original image
93
  image_for_display = self.resize_image(image_for_display, 500, 500)
94
  st.image(image_for_display)
95
+
 
96
 
97
  # Image uploader
98
  uploaded_image = self.col1.file_uploader("Or upload an Image", type=["png", "jpg", "jpeg"])
99
+
 
100
  ############################################
101
 
102