Fix: set cache dirs to avoid permission errors in Spaces
Browse files- app/model.py +5 -1
app/model.py
CHANGED
@@ -1,6 +1,10 @@
|
|
1 |
-
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
2 |
import os
|
|
|
|
|
|
|
|
|
3 |
import re
|
|
|
4 |
|
5 |
class CancerClassifier:
|
6 |
def __init__(self, model_path="user1729/BiomedBERT-cancer-bert-classifier-v1.0"):
|
|
|
|
|
1 |
import os
|
2 |
+
# Avoid cache write permission errors in Hugging Face Spaces
|
3 |
+
os.environ["HF_HOME"] = "/tmp/huggingface"
|
4 |
+
os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
|
5 |
+
os.environ["HF_DATASETS_CACHE"] = "/tmp/huggingface/datasets"
|
6 |
import re
|
7 |
+
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
8 |
|
9 |
class CancerClassifier:
|
10 |
def __init__(self, model_path="user1729/BiomedBERT-cancer-bert-classifier-v1.0"):
|