Spaces:

wondervictor
/

ControlAR

Running on Zero

wondervictor commited on 26 days ago

Commit

dfb2df7

•

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()