init

app.py

@@ -112,7 +112,7 @@ def segment_image(
 
 def get_depth_pipeline():
     feature_extractor = AutoImageProcessor.from_pretrained("LiheYoung/depth-anything-large-hf",
-                                                           torch_dtype=dtype).to(device)
+                                                           torch_dtype=dtype)
     depth_estimator = AutoModelForDepthEstimation.from_pretrained("LiheYoung/depth-anything-large-hf",
                                                                   torch_dtype=dtype).to(device)
     return feature_extractor, depth_estimator