Update app.py

app.py

@@ -13,7 +13,6 @@ 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 = """
@@ -90,7 +89,7 @@ with gr.Blocks(css=css) as demo:
 
         depth = predict_depth(image)
 
-        raw_depth = Image.fromarray(depth.astype('uint16'))
+        raw_depth = Image.fromarray(depth)
         tmp_raw_depth = tempfile.NamedTemporaryFile(suffix='.png', delete=False)
         raw_depth.save(tmp_raw_depth.name)