pan-yl commited on
Commit
3c9306c
1 Parent(s): 0060cd3
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -844,8 +844,8 @@ class ChatBotUI(object):
844
 
845
  img_num = len(edit_image) if edit_image is not None else 1
846
  imgs = self.pipe(
847
- input_image=edit_image,
848
- input_mask=edit_image_mask,
849
  task=edit_task,
850
  prompt=[prompt] * img_num,
851
  negative_prompt=[''] * img_num,
 
844
 
845
  img_num = len(edit_image) if edit_image is not None else 1
846
  imgs = self.pipe(
847
+ image=edit_image,
848
+ mask=edit_image_mask,
849
  task=edit_task,
850
  prompt=[prompt] * img_num,
851
  negative_prompt=[''] * img_num,