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"):
|