Update app.py

app.py

@@ -185,6 +185,7 @@ if __name__ == "__main__":
                 img_cv = rearrange(x_checked_image_torch[0], 'c h w -> h w c').cpu().numpy()
                 img_cv = (img_cv * 255).astype(np.uint8)
                 img_cv = cv2.cvtColor(img_cv, cv2.COLOR_RGB2BGR)
+                img_cv = cv2.resize(img_cv, (384, 512))
                 
                 #mask
                 mask_cv = mask_cv.convert("L").resize((384,512))