ariG23498 HF staff commited on
Commit
6cf04fa
1 Parent(s): 39354f2

turn mask into PIL

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -174,7 +174,7 @@ def run_open_gen_fill(image, edit_prompt):
174
  replace_with,
175
  caption,
176
  replaced_caption,
177
- masks,
178
  output,
179
  )
180
 
 
174
  replace_with,
175
  caption,
176
  replaced_caption,
177
+ Image.fromarray(masks[0][0][0, :, :].numpy()),
178
  output,
179
  )
180