dgoot commited on
Commit
a7a042b
·
verified ·
1 Parent(s): 3005122

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def infer(
46
  )
47
 
48
  init_image: Image.Image = image_editor["background"].convert("RGB")
49
- mask_image: Image.Image = image_editor["composite"].convert("RGB")
50
 
51
  # Downscale the image
52
  init_image.thumbnail((1024, 1024))
 
46
  )
47
 
48
  init_image: Image.Image = image_editor["background"].convert("RGB")
49
+ mask_image: Image.Image = image_editor["layers"][0].convert("RGB")
50
 
51
  # Downscale the image
52
  init_image.thumbnail((1024, 1024))