LPDoctor commited on
Commit
a333953
β€’
1 Parent(s): 4046cf6

resize the correct width and height

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