KB-VQA

Sleeping

m7mdal7aj commited on Feb 27, 2024

Commit

2ec2d03

verified ·

1 Parent(s): c327e49

Update my_model/tabs/run_inference.py

Files changed (1) hide show

my_model/tabs/run_inference.py CHANGED Viewed

@@ -96,7 +96,7 @@ class InferenceRunner(StateManager):
         # Image uploader
         uploaded_image = self.col1.file_uploader("Or upload an Image", type=["png", "jpg", "jpeg"])
-        image_for_display = copy.deepcopy(image_data['image'])  # resize just for the display control without changing original image
         image_for_display = self.resize_image(image_for_display, 500, 500)
 ############################################

         # Image uploader
         uploaded_image = self.col1.file_uploader("Or upload an Image", type=["png", "jpg", "jpeg"])
+        image_for_display = copy.deepcopy(uploaded_image)  # resize just for the display control without changing original image
         image_for_display = self.resize_image(image_for_display, 500, 500)
 ############################################