swayamshetkar commited on
Commit
97cb9c2
·
verified ·
1 Parent(s): 27dc146

Update utils/rmbg_model.py

Browse files
Files changed (1) hide show
  1. utils/rmbg_model.py +2 -0
utils/rmbg_model.py CHANGED
@@ -1,5 +1,7 @@
1
  import os
2
 
 
 
3
  # ✅ Use /tmp/hf_cache (always writable in Hugging Face Spaces)
4
  CACHE_DIR = "/tmp/hf_cache"
5
  os.environ["HF_HOME"] = CACHE_DIR
 
1
  import os
2
 
3
+ hf_token = os.getenv("HF_TOKEN")
4
+
5
  # ✅ Use /tmp/hf_cache (always writable in Hugging Face Spaces)
6
  CACHE_DIR = "/tmp/hf_cache"
7
  os.environ["HF_HOME"] = CACHE_DIR