vilarin commited on
Commit
f2fbc2b
·
verified ·
1 Parent(s): 116610e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -37,6 +37,9 @@ def inpaintGen(
37
  source_path = imgMask["background"]
38
  mask_path = imgMask["layers"][0]
39
 
 
 
 
40
  if not source_path:
41
  raise gr.Error("Please upload an image.")
42
 
@@ -44,7 +47,7 @@ def inpaintGen(
44
  raise gr.Error("Please draw a mask on the image.")
45
 
46
  source_img = load_image(source_path)
47
- mask_img = load_image(mask_path)
48
 
49
  width, height = source_img.size
50
 
 
37
  source_path = imgMask["background"]
38
  mask_path = imgMask["layers"][0]
39
 
40
+ print(f'source_path: {source_path}')
41
+ print(f'mask_path: {mask_path}')
42
+
43
  if not source_path:
44
  raise gr.Error("Please upload an image.")
45
 
 
47
  raise gr.Error("Please draw a mask on the image.")
48
 
49
  source_img = load_image(source_path)
50
+ mask_img = load_image(mask_path).split()[3]
51
 
52
  width, height = source_img.size
53