Update app.py

app.py

@@ -13,7 +13,7 @@ from gradio_imageslider import ImageSlider
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 from transformers import pipeline
-
+from PIL import Image
 from depth_anything_v2.dpt import DepthAnythingV2
 
 css = """
@@ -81,9 +81,12 @@ with gr.Blocks(css=css) as demo:
     cmap = matplotlib.colormaps.get_cmap('Spectral_r')
 
     def on_submit(image):
+        if isinstance(image, np.ndarray):
+            image = Image.fromarray(image)
+
         original_image = image.copy()
 
-        h, w = image.shape[:2]
+        h, w = image.size  # For PIL images, use .size instead of .shape
 
         depth = predict_depth(image)