Update app.py

app.py

@@ -81,8 +81,13 @@ def process(input_image,
     randomize_seed, 
     guidance_scale, 
     num_inference_steps):
+    
     if original_image is None:
-        raise gr.Error('Please upload the input image')
+        if input_image:
+            original_image = input_image
+            original_mask = input_mask
+        else:
+            raise gr.Error('Please upload the input image')
     if (original_mask is None or len(selected_points)==0) and input_mask is None:
         raise gr.Error("Please click the region where you hope unchanged/changed, or upload a white-black Mask image")
     
@@ -176,8 +181,8 @@ with block:
                 with gr.Row():
                     with gr.Tabs(elem_classes=["feedback"]):
                         with gr.TabItem("Input Image"):
-                            input_image = gr.Image(type="numpy", label="input",scale=2, height=640)
-                original_image = gr.State(value=None,label="index")
+                            input_image = gr.Image(type="numpy", label="input",scale=2, height=1024)
+                original_image = gr.State(value=None, label="index")
                 original_mask = gr.State(value=None)
                 selected_points = gr.State([],label="select points")
                 with gr.Row(elem_id="Seg"):
@@ -225,7 +230,7 @@ with block:
                 with gr.Row(elem_id="Image"):
                     with gr.Tabs(elem_classes=["feedback1"]):
                         with gr.TabItem("User-specified Mask Image (Optional)"):
-                            input_mask = gr.Image(type="numpy", label="Mask Image", height=640)
+                            input_mask = gr.Image(type="numpy", label="Mask Image", height=1024)
             
         with gr.Column():
             with gr.Tabs(elem_classes=["feedback"]):
@@ -237,7 +242,7 @@ with block:
         example = gr.Examples(
             label="Input Example",
             examples=image_examples,
-            inputs=[input_image, prompt, input_mask, original_image, selected_points,result_gallery],
+            inputs=[input_image, prompt, input_mask, original_image, selected_points, result_gallery],
             outputs=[input_image, prompt, input_mask, original_image, selected_points],
             fn=process_example,
             run_on_click=True,