Spaces:

cborg
/

imgprivllm

Sleeping

hugohabicht01 commited on Apr 21

Commit

53dd78d

1 Parent(s): 6aa2d3f

fix sam cfg not being available

Files changed (1) hide show

blurnonymize.py CHANGED Viewed

@@ -151,8 +151,8 @@ class ImageBlurnonymizer:
         self.device = None
         self.model_cfg = None
-        self.checkpoint_name = "./sam2.1_hiera_small.pt"
-        self.model_cfg_name = "./sam2.1_hiera_s.yaml"
         self.init_sam()
     def init_sam(self, force=False):

         self.device = None
         self.model_cfg = None
+        self.checkpoint_name = "/data/sam2.1_hiera_small.pt"
+        self.model_cfg_name = "/data/sam2.1_hiera_s.yaml"
         self.init_sam()
     def init_sam(self, force=False):