Update app.py

app.py

@@ -130,12 +130,12 @@ def start_tryon(person_img, pose_img, mask_img, cloth_img, garment_des, denoise_
     pipe.unet_encoder.to(device)
 
     # 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))
+    garm_img = cloth_img.convert("RGB").resize((768, 1024)).to(device, torch.float16)
+    human_img = person_img.convert("RGB").resize((768, 1024)).to(device, torch.float16)
+    mask = mask_img.convert("RGB").resize((768, 1024)).to(device, torch.float16)
 
     # Prepare pose image (already uploaded)
-    pose_img = pose_img.resize((768, 1024))
+    pose_img = pose_img.resize((768, 1024)).to(device, torch.float16)
 
     
     
@@ -196,11 +196,11 @@ 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,
-                image=human_img,
+                mask_image=mask_tensor.to(device, torch.float16),
+                image=human_img.to(device, torch.float16),
                 height=1024,
                 width=768,
-                ip_adapter_image=garm_img.resize((768, 1024)),
+                ip_adapter_image=garm_img.resize((768, 1024)).to(device, torch.float16),
                 guidance_scale=2.0,
             )[0]