Update app.py

app.py

@@ -2,7 +2,7 @@ import torch
 torch.jit.script = lambda f: f
 from zoedepth.utils.misc import colorize, save_raw_16bit
 from zoedepth.utils.geometry import depth_to_points, create_triangles
-from diffusers import DiffusionPipeline
+from marigold_depth_estimation import MarigoldPipeline
 import gradio as gr
 import spaces
 
@@ -31,7 +31,7 @@ DEVICE = 'cuda'
 model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).to("cpu").eval()
 
 CHECKPOINT = "prs-eth/marigold-v1-0"
-pipe = DiffusionPipeline.from_pretrained(CHECKPOINT)
+pipe = MarigoldPipeline.from_pretrained(CHECKPOINT)
 
 # ----------- Depth functions
 @spaces.GPU(enable_queue=True)