tonyassi commited on
Commit
58580be
·
verified ·
1 Parent(s): b77bbcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ def inpaint(ip, image_editor, prompt, neg_prompt, width, height, ip_scale, stren
68
  inpaint_pipeline.to("cuda")
69
  print(image_editor)
70
  image = image_editor['background'].convert('RGB')
71
- mask = image_editor['composite'].convert('RGB')
72
  mask = ImageOps.invert(mask)
73
 
74
  ip.thumbnail((1024, 1024))
 
68
  inpaint_pipeline.to("cuda")
69
  print(image_editor)
70
  image = image_editor['background'].convert('RGB')
71
+ mask = image_editor["layers"][0].convert('RGB')
72
  mask = ImageOps.invert(mask)
73
 
74
  ip.thumbnail((1024, 1024))