Update preprocessor.py

preprocessor.py

@@ -9,7 +9,7 @@ from controlnet_aux import (
     HEDdetector,
     # LineartAnimeDetector,
     LineartDetector,
-    # MidasDetector,
+    MidasDetector,
     # MLSDdetector,
     # NormalBaeDetector,
     # OpenposeDetector,
@@ -74,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()