Update app.py

app.py

@@ -43,10 +43,11 @@ def process_img(image):
     img = img / 255.
     img = img.astype(np.float32)
     
-    resize = transforms.Resize((720, 1280))
-    img = resize(img)
+
        
     y = torch.tensor(img).permute(2,0,1).unsqueeze(0).to(device)
+    resize = transforms.Resize((720, 1280))
+    y = resize(y)
 
     with torch.no_grad():
         x_hat = model(y)