Update interaction code

app.py

@@ -43,7 +43,7 @@ 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):
+def image_preprocess(image: gr.Image):
     if image is None:
         return None, None
 
@@ -51,13 +51,6 @@ def save_original_image(image: gr.Image):
     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
-
     PROCESSOR.image_processor.do_normalize = False
     image_tensor = PROCESSOR(images=[image], return_tensors="pt", padding=True)
     PROCESSOR.image_processor.do_normalize = True
@@ -87,9 +80,8 @@ with gr.Blocks(analytics_enabled=True, title=PAPER_TITLE, theme="soft") as demo:
     gr.Markdown(MARKDOWN_DESCRIPTION)
     with gr.Row():
         with gr.Column():
-            curr_image = gr.Image(label="input", type="pil")
-            _orig_image = gr.Image(
-                label="orig. image", type="pil", visible=False, interactive=False
+            curr_image = gr.Image(
+                label="input", type="pil", sources=["upload", "webcam", "clipboard"]
             )
             alpha_slider = gr.Slider(0.0, 1.0, value=0.7, step=0.1, label="alpha")
             with gr.Row():
@@ -97,9 +89,13 @@ with gr.Blocks(analytics_enabled=True, title=PAPER_TITLE, theme="soft") as demo:
                 run_button = gr.Button(value="Submit", variant="primary")
         with gr.Column():
             output_label = gr.Label(label="output", num_top_classes=5)
+
+    _orig_image = gr.Image(label="original image", type="pil", visible=False, interactive=False)
+    _example_image = gr.Image(label="example image", type="pil", visible=False, interactive=False)
+
     examples = gr.Examples(
         examples=glob(os.path.join(os.path.dirname(__file__), "examples", "*.jpg")),
-        inputs=[_orig_image],
+        inputs=[_example_image],
         outputs=[output_label],
         fn=image_inference,
         cache_examples=True,
@@ -108,10 +104,10 @@ with gr.Blocks(analytics_enabled=True, title=PAPER_TITLE, theme="soft") as demo:
 
     # INTERACTIONS
     # - change
-    _orig_image.change(prepare_image, [_orig_image], [curr_image, _orig_image])
+    _example_image.change(image_preprocess, [_example_image], [curr_image, _orig_image])
 
     # - upload
-    curr_image.upload(save_original_image, [curr_image], [curr_image, _orig_image])
+    curr_image.upload(image_preprocess, [curr_image], [curr_image, _orig_image])
     curr_image.upload(lambda: None, [], [output_label])
 
     # - clear