Update app.py

app.py

@@ -129,6 +129,21 @@ with gr.Blocks() as demo:
         with gr.Column():
             result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery", preview=True)
 
+    # once user upload an image, the original image is stored in `original_image`
+    def store_img(img):
+        # image upload is too slow
+        if min(img.shape[0], img.shape[1]) > 1024:
+            img = resize_image(img, 1024)
+        if max(img.shape[0], img.shape[1])*1.0/min(img.shape[0], img.shape[1])>2.0:
+            raise gr.Error('image aspect ratio cannot be larger than 2.0')
+        return img, img, [], None  # when new image is uploaded, `selected_points` should be empty
+
+    original_image.upload(
+        store_img,
+        [original_image],
+        [input_image, original_image, selected_points]
+    )
+    
     inputs = [original_image, input_mask, prompt, negative_prompt, blended, invert_mask, control_strength, seed, randomize_seed, guidance_scale, num_inference_steps]
     run_button.click(fn=process, inputs=inputs, outputs=[result_gallery])