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)
|