Keshabwi66 commited on
Commit
31b5073
·
verified ·
1 Parent(s): d998c35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -139,7 +139,7 @@ def start_tryon(person_img,cloth_img, garment_des, denoise_steps=10, seed=42):
139
  if is_checked:
140
  keypoints = openpose_model(human_img.resize((384,512)))
141
  model_parse, _ = parsing_model(human_img.resize((384,512)))
142
- mask= get_mask_location('hd', "upper_body", model_parse, keypoints)
143
  mask = mask.resize((768,1024))
144
 
145
  #mask = pil_to_binary_mask(mask.convert("RGB").resize((768, 1024)))
 
139
  if is_checked:
140
  keypoints = openpose_model(human_img.resize((384,512)))
141
  model_parse, _ = parsing_model(human_img.resize((384,512)))
142
+ mask, mask_gray= get_mask_location('hd', "upper_body", model_parse, keypoints)
143
  mask = mask.resize((768,1024))
144
 
145
  #mask = pil_to_binary_mask(mask.convert("RGB").resize((768, 1024)))