CRYPTONEWS34 commited on
Commit
1eee1c0
·
1 Parent(s): a3bfc0a

Fix: Use HF_HOME=/data/huggingface to avoid permission error

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. app.py +2 -1
Dockerfile CHANGED
@@ -1,6 +1,6 @@
1
  FROM python:3.10-slim
2
  RUN mkdir -p /tmp/huggingface
3
-
4
  # Working directory
5
  WORKDIR /app
6
 
 
1
  FROM python:3.10-slim
2
  RUN mkdir -p /tmp/huggingface
3
+ ENV HF_HOME=/tmp
4
  # Working directory
5
  WORKDIR /app
6
 
app.py CHANGED
@@ -2,13 +2,14 @@ import os
2
 
3
  os.environ["HF_HOME"] = "/tmp/huggingface"
4
  os.environ["MPLCONFIGDIR"] = "/tmp/mplconfig"
5
- os.environ["TRANSFORMERS_CACHE"] = "./hf_cache"
6
 
7
  os.makedirs("/tmp/huggingface", exist_ok=True)
8
  os.makedirs("/tmp/mplconfig", exist_ok=True)
9
 
10
  from transformers import AutoTokenizer, AutoModelForTokenClassification
11
 
 
12
  # Load the NER model
13
  tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
14
  model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
 
2
 
3
  os.environ["HF_HOME"] = "/tmp/huggingface"
4
  os.environ["MPLCONFIGDIR"] = "/tmp/mplconfig"
5
+ os.environ["HF_HOME"] = "/tmp"
6
 
7
  os.makedirs("/tmp/huggingface", exist_ok=True)
8
  os.makedirs("/tmp/mplconfig", exist_ok=True)
9
 
10
  from transformers import AutoTokenizer, AutoModelForTokenClassification
11
 
12
+ os.environ["HF_HOME"] = "/tmp"
13
  # Load the NER model
14
  tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
15
  model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")