Update app.py

app.py

@@ -132,7 +132,7 @@ def start_tryon(person_img, pose_img, mask_img, cloth_img, garment_des, denoise_
     # Resize and prepare images
     garm_img = cloth_img.convert("RGB").resize((768, 1024))
     human_img = person_img.convert("RGB").resize((768, 1024))
-    mask = mask_img.convert("RGB").resize((768, 1024))
+    mask = pil_to_binary_mask(mask_img.convert("RGB").resize((768, 1024)))
 
     # Prepare pose image (already uploaded)
     pose_img = pose_img.resize((768, 1024))
@@ -179,7 +179,6 @@ def start_tryon(person_img, pose_img, mask_img, cloth_img, garment_des, denoise_
             # Convert images to tensors for processing
             pose_img_tensor = tensor_transfrom(pose_img).unsqueeze(0).to(device, torch.float16)
             garm_tensor = tensor_transfrom(garm_img).unsqueeze(0).to(device, torch.float16)
-            mask_tensor = tensor_transfrom(mask).unsqueeze(0).to(device, torch.float16)
 
             # Prepare the generator with optional seed
             generator = torch.Generator(device).manual_seed(seed) if seed is not None else None
@@ -196,7 +195,7 @@ def start_tryon(person_img, pose_img, mask_img, cloth_img, garment_des, denoise_
                 pose_img=pose_img_tensor.to(device, torch.float16),
                 text_embeds_cloth=prompt_embeds_cloth.to(device, torch.float16),
                 cloth=garm_tensor.to(device, torch.float16),
-                mask_image=mask_tensor.to(device, torch.float16),
+                mask_image=mask,
                 image=human_img,
                 height=1024,
                 width=768,