Update app.py

app.py

@@ -42,13 +42,14 @@ def process_img(image):
     img = np.array(image)
     img = img / 255.
     img = img.astype(np.float32)
+    
+    resize = transforms.Resize((720, 1280))
+    mask_rgb = resize(mask_rgb)
+       
     y = torch.tensor(img).permute(2,0,1).unsqueeze(0).to(device)
-    _, _, H, W = y.shape
-    y= check_image_size(y)
 
     with torch.no_grad():
         x_hat = model(y)
-        x_hat = x_hat[:, :, :H, :W]
 
     restored_img = x_hat.squeeze().permute(1,2,0).clamp_(0, 1).cpu().detach().numpy()
     restored_img = np.clip(restored_img, 0. , 1.)