Update app.py

app.py

@@ -9,6 +9,7 @@ DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
 model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).to(DEVICE).eval()
 
 def predict(image):
+    image.thumbnail((1024,1024))
     depth = model.infer_pil(image)
     colored_depth, vmin, vmax = colorize(depth, cmap='gray_r')