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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -140,8 +140,7 @@ def start_tryon(person_img,cloth_img, garment_des, denoise_steps=10, seed=42):
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
- return mask
144
- mask = mask.resize((768,1024))
145
 
146
  #mask = pil_to_binary_mask(mask.convert("RGB").resize((768, 1024)))
147
  pose_img=Image.open("00006_00.jpg")
 
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)))
146
  pose_img=Image.open("00006_00.jpg")