Update app.py

app.py

@@ -129,6 +129,7 @@ pipe.unet_encoder = UNet_Encoder
 @spaces.GPU
 def start_tryon(boy,girl,person_img,cloth_img, garment_des, denoise_steps=10, seed=42):
     # Assuming device is set up (e.g., "cuda" or "cpu")
+    
     device="cuda"
     openpose_model.preprocessor.body_estimation.model.to(device)
     pipe.to(device)
@@ -223,7 +224,7 @@ def start_tryon(boy,girl,person_img,cloth_img, garment_des, denoise_steps=10, se
                 guidance_scale=2.0,
             )[0]
 
-    return images[0]
+    return images[0].resize(person_img.shape)
 
 
 # Gradio interface for the virtual try-on model