add size check

app.py

@@ -32,6 +32,10 @@ def process_images(image, image2=None,inpaint_radius=3,blur_radius=25,edge_expan
               return image
          mask = image['layers'][0]
 
+    img_width,img_height = image["background"].size
+    mask_width, mask_height = mask.size
+    if img_width!=mask_width or img_height!=mask_height:
+         gr.Error(f"image size({img_width},{img_height}) must be same as mask size({mask_width},{mask_height})")
 
     output = opencvinpaint.process_cvinpaint(image["background"],mask,inpaint_radius,blur_radius,edge_expand,inpaint_mode)
         
@@ -65,7 +69,7 @@ with gr.Blocks(css=css, elem_id="demo-container") as demo:
         gr.HTML(read_file("demo_header.html"))
     with gr.Row():
                 with gr.Column():
-                    image = gr.ImageEditor(sources=['upload','clipboard'],transforms=[],image_mode='RGB', layers=False,  elem_id="image_upload", type="pil", label="Upload",brush=gr.Brush(colors=["#fff"], color_mode="fixed"))
+                    image = gr.ImageEditor(height=800,sources=['upload','clipboard'],transforms=[],image_mode='RGB', layers=False,  elem_id="image_upload", type="pil", label="Upload",brush=gr.Brush(colors=["#fff"], color_mode="fixed"))
                     with gr.Row(elem_id="prompt-container",  equal_height=False):
                         with gr.Row():
                             btn = gr.Button("Inpaint!", elem_id="run_button")
@@ -79,7 +83,7 @@ with gr.Blocks(css=css, elem_id="demo-container") as demo:
                             modes = ["Telea", "Navier-Stokes"]
                             inpaint_mode = gr.Dropdown(label="modes", choices=modes, value="Telea") 
                 with gr.Column():
-                    image_out = gr.Image(sources=[],label="Output", elem_id="output-img")
+                    image_out = gr.Image(height=800,sources=[],label="Output", elem_id="output-img")
                     mask_out = gr.Image(height=800,sources=[],label="Mask", elem_id="mask-img",format="jpeg")