Avoid double `prepare_image` when uploading

app.py

@@ -4,7 +4,7 @@ from typing import Optional
 
 import gradio as gr
 import torch
-from torchvision.transforms.functional import to_pil_image
+from torchvision.transforms.functional import resize, to_pil_image
 from transformers import AutoModel, CLIPProcessor
 
 PAPER_TITLE = "Vocabulary-free Image Classification"
@@ -36,6 +36,17 @@ MODEL = AutoModel.from_pretrained("altndrr/cased", trust_remote_code=True).to(DE
 PROCESSOR = CLIPProcessor.from_pretrained("openai/clip-vit-large-patch14")
 
 
+def save_original_image(image: gr.Image):
+    if image is None:
+        return None, None
+
+    size = PROCESSOR.image_processor.size["shortest_edge"]
+    size = min(size) if isinstance(size, tuple) else size
+    image = resize(image, size)
+
+    return image, image.copy()
+
+
 def prepare_image(image: gr.Image):
     if image is None:
         return None, None
@@ -93,7 +104,7 @@ with gr.Blocks(analytics_enabled=True, title=PAPER_TITLE, theme="soft") as demo:
     _orig_image.change(prepare_image, [_orig_image], [curr_image, _orig_image])
 
     # - upload
-    curr_image.upload(prepare_image, [curr_image], [curr_image, _orig_image])
+    curr_image.upload(save_original_image, [curr_image], [curr_image, _orig_image])
     curr_image.upload(lambda: None, [], [output_label])
 
     # - clear