hassoudi commited on
Commit
38e01ae
·
verified ·
1 Parent(s): 9dcc0d8

Upload model.py

Browse files

fix model.py for env vars

Files changed (1) hide show
  1. model.py +4 -0
model.py CHANGED
@@ -1,7 +1,11 @@
1
  import os
2
  from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
3
 
 
4
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
 
 
 
5
 
6
 
7
  HF_TOKEN = os.getenv("HF_TOKEN")
 
1
  import os
2
  from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
3
 
4
+ # Safe writable cache dirs in Hugging Face Spaces
5
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
6
+ os.environ["HF_HOME"] = "/tmp/hf_home"
7
+ os.environ["HF_HUB_CACHE"] = "/tmp/hf_hub"
8
+
9
 
10
 
11
  HF_TOKEN = os.getenv("HF_TOKEN")