m7mdal7aj commited on
Commit
3b58975
1 Parent(s): 436f3b4

Update my_model/tabs/run_inference.py

Browse files
Files changed (1) hide show
  1. my_model/tabs/run_inference.py +2 -1
my_model/tabs/run_inference.py CHANGED
@@ -35,7 +35,8 @@ class InferenceRunner(StateManager):
35
  for idx, sample_image_path in enumerate(self.sample_images):
36
  with cols[idx]:
37
  image = Image.open(sample_image_path)
38
- st.image(image, width=100, height=100)
 
39
  if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx}'):
40
  self.process_new_image(sample_image_path, image, kbvqa)
41
 
 
35
  for idx, sample_image_path in enumerate(self.sample_images):
36
  with cols[idx]:
37
  image = Image.open(sample_image_path)
38
+ image = self.resize_image(sample_image_path, new_width, new_height)
39
+ st.image(image)
40
  if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx}'):
41
  self.process_new_image(sample_image_path, image, kbvqa)
42