Update preprocessor.py

preprocessor.py

@@ -18,7 +18,7 @@ from controlnet_aux import (
 from controlnet_aux.util import HWC3
 from transformers import pipeline
 # from cv_utils import resize_image
-# from depth_estimator import DepthEstimator
+from depth_estimator import DepthEstimator
 
 
 class DepthEstimator:
@@ -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()