Spaces:

swayamshetkar
/

Removeit

Sleeping

swayamshetkar commited on 28 days ago

Commit

e96f8ef

verified ·

1 Parent(s): 97cb9c2

Update utils/rmbg_model.py

Files changed (1) hide show

utils/rmbg_model.py CHANGED Viewed

@@ -32,7 +32,8 @@ class RMBGRemover:
             "briaai/RMBG-1.4",
             trust_remote_code=True,
             revision="2ceba5a5efaec153162aedea169f76caf9b46cf8",
-            cache_dir=CACHE_DIR
         ).to(self.device)
         self.model.eval()

             "briaai/RMBG-1.4",
             trust_remote_code=True,
             revision="2ceba5a5efaec153162aedea169f76caf9b46cf8",
+            cache_dir=CACHE_DIR,
+            token=hf_token
         ).to(self.device)
         self.model.eval()