LPDoctor commited on
Commit
3417909
β€’
1 Parent(s): 17bbdd8

only return generated image

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -237,9 +237,9 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
237
  if is_checked_crop:
238
  out_img = images[0].resize(crop_size)
239
  human_img_orig.paste(out_img, (int(left), int(top)))
240
- return human_img_orig, mask_gray
241
  else:
242
- return images[0], mask_gray
243
  # return images[0], mask_gray
244
 
245
  garm_list = os.listdir(os.path.join(example_path,"cloth"))
 
237
  if is_checked_crop:
238
  out_img = images[0].resize(crop_size)
239
  human_img_orig.paste(out_img, (int(left), int(top)))
240
+ return human_img_orig
241
  else:
242
+ return images[0]
243
  # return images[0], mask_gray
244
 
245
  garm_list = os.listdir(os.path.join(example_path,"cloth"))