Update app.py

app.py

@@ -33,6 +33,7 @@ def predict(input_image, prompt, negative_prompt, steps, num_samples, scale, see
     depth = None
     if depth_image is not None:
         depth_image = pad_image(depth_image)
+        depth_image = depth_image.resize((512, 512))
         depth = np.array(depth_image.convert("L"))
         depth = depth.astype(np.float32) / 255.0
         depth = depth[None, None]