Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -134,13 +134,15 @@ def start_tryon(person_img,cloth_img, garment_des, denoise_steps=10, seed=42):
|
|
134 |
# Resize and prepare images
|
135 |
garm_img = cloth_img.convert("RGB").resize((768, 1024))
|
136 |
human_img = person_img.convert("RGB").resize((768, 1024))
|
|
|
137 |
is_checked=True;
|
138 |
if is_checked:
|
139 |
keypoints = openpose_model(human_img.resize((384,512)))
|
140 |
model_parse, _ = parsing_model(human_img.resize((384,512)))
|
141 |
mask= get_mask_location('hd', "upper_body", model_parse, keypoints)
|
|
|
142 |
|
143 |
-
mask = pil_to_binary_mask(mask.convert("RGB").resize((768, 1024)))
|
144 |
pose_img=Image.open("00006_00.jpg")
|
145 |
|
146 |
# Prepare pose image (already uploaded)
|
|
|
134 |
# Resize and prepare images
|
135 |
garm_img = cloth_img.convert("RGB").resize((768, 1024))
|
136 |
human_img = person_img.convert("RGB").resize((768, 1024))
|
137 |
+
|
138 |
is_checked=True;
|
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)))
|
146 |
pose_img=Image.open("00006_00.jpg")
|
147 |
|
148 |
# Prepare pose image (already uploaded)
|