Update preprocessor.py

preprocessor.py

@@ -23,7 +23,7 @@ from transformers import pipeline
 
 class DepthEstimator:
     def __init__(self):
-        self.model = pipeline("condition/ckpts/dpt_large")
+        self.model = pipeline("depth-estimation")
 
     def __call__(self, image: np.ndarray, **kwargs) -> PIL.Image.Image:
         detect_resolution = kwargs.pop("detect_resolution", 512)
@@ -55,7 +55,8 @@ def resize_image(input_image, resolution, interpolation=None):
 
 
 class Preprocessor:
-    MODEL_ID = "condition/ckpts"
+    # MODEL_ID = "condition/ckpts"
+    MODEL_ID = "lllyasviel/Annotators"
 
     def __init__(self):
         self.model = None
@@ -73,8 +74,8 @@ class Preprocessor:
         elif name == "Canny":
             self.model = CannyDetector()
         elif name == "Depth":
-            # self.model = DepthEstimator()
-            self.model = MidasDetector.from_pretrained(self.MODEL_ID)
+            self.model = DepthEstimator()
+            # self.model = MidasDetector.from_pretrained(self.MODEL_ID)
         else:
             raise ValueError
         torch.cuda.empty_cache()