sync the output size as same as input

app.py

@@ -129,6 +129,8 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
     pipe.to(device)
     pipe.unet_encoder.to(device)
 
+    OUTPUT_WIDTH, OUTPUT_HEIGHT = dict.size
+    
     garm_img= garm_img.convert("RGB").resize((768,1024))
     
     human_img_orig = dict.convert("RGB").resize((768,1024))
@@ -238,9 +240,9 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
     if is_checked_crop:
         out_img = images[0].resize(crop_size)        
         human_img_orig.paste(out_img, (int(left), int(top)))    
-        return human_img_orig
+        return human_img_orig.resize(OUTPUT_WIDTH, OUTPUT_HEIGHT)
     else:
-        return images[0]
+        return images[0].resize(OUTPUT_WIDTH, OUTPUT_HEIGHT)
     # return images[0], mask_gray
 
 garm_list = os.listdir(os.path.join(example_path,"cloth"))