chaojiemao commited on
Commit
066c1c0
·
verified ·
1 Parent(s): ceaed92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -454,7 +454,7 @@ class DemoUI(object):
454
 
455
  edit_image = self.construct_edit_image(edit_image, edit_mask)
456
 
457
- pre_edit_image, pre_edit_mask, pre_ref_image = preprocess_input(ref_image, edit_image)
458
  pre_edit_image = edit_preprocess(edit_info, we.device_id, pre_edit_image, pre_edit_mask)
459
  edit_info = edit_info or {}
460
  repainting_scale = edit_info.get("REPAINTING_SCALE", 1.0)
 
454
 
455
  edit_image = self.construct_edit_image(edit_image, edit_mask)
456
 
457
+ pre_edit_image, pre_edit_mask, pre_ref_image, is_suc, err_msg = preprocess_input(ref_image, edit_image)
458
  pre_edit_image = edit_preprocess(edit_info, we.device_id, pre_edit_image, pre_edit_mask)
459
  edit_info = edit_info or {}
460
  repainting_scale = edit_info.get("REPAINTING_SCALE", 1.0)