Update app.py

app.py

@@ -12,6 +12,7 @@ import tempfile
 from gradio_imageslider import ImageSlider
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
+from transformers import pipeline
 
 from depth_anything_v2.dpt import DepthAnythingV2
 
@@ -49,10 +50,12 @@ filepath = hf_hub_download(repo_id="depth-anything/Depth-Anything-V2-Metric-Indo
 
 # Convert to PyTorch tensor
 #state_dict = torch.load(filepath, map_location="cpu", weights_only=True)
-state_dict = load_file(filepath)
+#state_dict = load_file(filepath)
 
-model.load_state_dict(state_dict)
-model = model.to(DEVICE).eval()
+#model.load_state_dict(state_dict)
+#model = model.to(DEVICE).eval()
+
+pipe = pipeline(task="depth-estimation", model="depth-anything/Depth-Anything-V2-Metric-Indoor-Large-hf")
 
 title = "# Depth Anything V2"
 description = """Official demo for **Depth Anything V2**.
@@ -60,7 +63,8 @@ Please refer to our [paper](https://arxiv.org/abs/2406.09414), [project page](ht
 
 @spaces.GPU
 def predict_depth(image):
-    return model.infer_image(image)
+    #return model.infer_image(image)
+    return pipe(image)["depth"]
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown(title)