Spaces:

cborg
/

imgprivllm

Running on Zero

hugohabicht01 commited on Apr 21

Commit

46161a2

1 Parent(s): cce3710

fix??

Files changed (1) hide show

blurnonymize.py CHANGED Viewed

@@ -162,7 +162,7 @@ class ImageBlurnonymizer:
         self.download_weights()
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
-        sam = build_sam2(self.model_cfg_name, self.checkpoint_name, device=self.device)
         self.predictor = SAM2ImagePredictor(sam)
     def download_weights(self):

         self.download_weights()
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
+        sam = build_sam2(f"file://{self.model_cfg_name}", f"file://{self.checkpoint_name}", device=self.device)
         self.predictor = SAM2ImagePredictor(sam)
     def download_weights(self):