Keshabwi66 commited on
Commit
7debb98
·
1 Parent(s): 79620b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -130,12 +130,12 @@ def start_tryon(person_img, pose_img, mask_img, cloth_img, garment_des, denoise_
130
  pipe.unet_encoder.to(device)
131
 
132
  # Resize and prepare images
133
- garm_img = cloth_img.convert("RGB").resize((768, 1024)).to(device, torch.float16)
134
- human_img = person_img.convert("RGB").resize((768, 1024)).to(device, torch.float16)
135
- mask = mask_img.convert("RGB").resize((768, 1024)).to(device, torch.float16)
136
 
137
  # Prepare pose image (already uploaded)
138
- pose_img = pose_img.resize((768, 1024)).to(device, torch.float16)
139
 
140
 
141
 
 
130
  pipe.unet_encoder.to(device)
131
 
132
  # Resize and prepare images
133
+ garm_img = cloth_img.convert("RGB").resize((768, 1024))
134
+ human_img = person_img.convert("RGB").resize((768, 1024))
135
+ mask = mask_img.convert("RGB").resize((768, 1024))
136
 
137
  # Prepare pose image (already uploaded)
138
+ pose_img = pose_img.resize((768, 1024))
139
 
140
 
141