Keshabwi66 commited on
Commit
2b1ec5b
·
verified ·
1 Parent(s): 348d6f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -224,7 +224,7 @@ def start_tryon(boy,girl,person_img,cloth_img, garment_des, denoise_steps=10, se
224
  guidance_scale=2.0,
225
  )[0]
226
 
227
- return images[0].resize(person_img.shape)
228
 
229
 
230
  # Gradio interface for the virtual try-on model
 
224
  guidance_scale=2.0,
225
  )[0]
226
 
227
+ return images[0].resize(person_img.size)
228
 
229
 
230
  # Gradio interface for the virtual try-on model