Spaces:
Running
on
Zero
Running
on
Zero
hugohabicht01
commited on
Commit
·
a07c49e
1
Parent(s):
6487a82
maybe fix?
Browse files- blurnonymize.py +1 -1
blurnonymize.py
CHANGED
|
@@ -157,7 +157,7 @@ class ImageBlurnonymizer:
|
|
| 157 |
return
|
| 158 |
|
| 159 |
self.device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 160 |
-
self.predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-large")
|
| 161 |
|
| 162 |
@staticmethod
|
| 163 |
def _smoothen_mask(mask: np.ndarray):
|
|
|
|
| 157 |
return
|
| 158 |
|
| 159 |
self.device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 160 |
+
self.predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-large", device=self.device)
|
| 161 |
|
| 162 |
@staticmethod
|
| 163 |
def _smoothen_mask(mask: np.ndarray):
|