Spaces:
Saad0KH
/
Running on Zero

Saad0KH commited on
Commit
c761dc5
ยท
verified ยท
1 Parent(s): bb5afb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -149,7 +149,7 @@ def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denois
149
  if is_checked:
150
  keypoints = openpose_model(human_img.resize((384, 512)))
151
  model_parse, _ = parsing_model(human_img.resize((384, 512)))
152
- mask, mask_gray = get_mask_location('hd', "upper_body", model_parse, keypoints)
153
  mask = mask.resize((768, 1024))
154
  else:
155
  mask = pil_to_binary_mask(dict['layers'][0].convert("RGB").resize((768, 1024)))
 
149
  if is_checked:
150
  keypoints = openpose_model(human_img.resize((384, 512)))
151
  model_parse, _ = parsing_model(human_img.resize((384, 512)))
152
+ mask, mask_gray = get_mask_location('hd', "full_body", model_parse, keypoints)
153
  mask = mask.resize((768, 1024))
154
  else:
155
  mask = pil_to_binary_mask(dict['layers'][0].convert("RGB").resize((768, 1024)))