Update app.py

app.py

@@ -44,7 +44,7 @@ def process_img(image):
     img = img.astype(np.float32)
     
     resize = transforms.Resize((720, 1280))
-    mask_rgb = resize(mask_rgb)
+    img = resize(img)
        
     y = torch.tensor(img).permute(2,0,1).unsqueeze(0).to(device)