Update app.py

app.py

@@ -133,14 +133,15 @@ def start_tryon(person_img,cloth_img, garment_des, denoise_steps=10, seed=42):
 
     # Resize and prepare images
     garm_img = cloth_img.convert("RGB").resize((768, 1024))
-    human_img = person_img.convert("RGB").resize((768, 1024))
+    human_img = person_img.convert("RGB")
 
     is_checked=True;
     if is_checked:
         keypoints = openpose_model(human_img.resize((384,512)))
         model_parse, _ = parsing_model(human_img.resize((384,512)))
         mask= get_mask_location('hd', "upper_body", model_parse, keypoints)
-        mask = mask.resize((768,1024))
+        return mask
+    mask = mask.resize((768,1024))
 
     #mask = pil_to_binary_mask(mask.convert("RGB").resize((768, 1024)))
     pose_img=Image.open("00006_00.jpg")