Akjava commited on
Commit
c3ff188
1 Parent(s): 8a56dc6

add size check

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -36,6 +36,9 @@ def process_images(image, image2=None,inpaint_radius=3,blur_radius=25,edge_expan
36
  mask_width, mask_height = mask.size
37
  if img_width!=mask_width or img_height!=mask_height:
38
  gr.Error(f"image size({img_width},{img_height}) must be same as mask size({mask_width},{mask_height})")
 
 
 
39
 
40
  output = opencvinpaint.process_cvinpaint(image["background"],mask,inpaint_radius,blur_radius,edge_expand,inpaint_mode)
41
 
 
36
  mask_width, mask_height = mask.size
37
  if img_width!=mask_width or img_height!=mask_height:
38
  gr.Error(f"image size({img_width},{img_height}) must be same as mask size({mask_width},{mask_height})")
39
+ else:
40
+ pass
41
+ print("size ok")
42
 
43
  output = opencvinpaint.process_cvinpaint(image["background"],mask,inpaint_radius,blur_radius,edge_expand,inpaint_mode)
44