Spaces:

wondervictor
/

ControlAR

Runtime error

wondervictor commited on Dec 13, 2024

Commit

dfb2df7

verified ·

1 Parent(s): 4038a0d

Update preprocessor.py

Files changed (1) hide show

preprocessor.py CHANGED Viewed

@@ -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)
         else:
             raise ValueError
         torch.cuda.empty_cache()

         elif name == "Canny":
             self.model = CannyDetector()
         elif name == "Depth":
+            # self.model = DepthEstimator()
+            self.model = MidasDetector.from_pretrained(self.MODEL_ID)
         else:
             raise ValueError
         torch.cuda.empty_cache()