fix cache permission using HF_HOME=/home/user/.cache/huggingface

Dockerfile

@@ -9,8 +9,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 EXPOSE 7860
 
-ENV HF_HOME=/app/hf_cache
-ENV TRANSFORMERS_CACHE=/app/hf_cache
-RUN mkdir -p /app/hf_cache
+ENV HF_HOME=/home/user/.cache/huggingface
+RUN mkdir -p /home/user/.cache/huggingface
 
 CMD ["python", "main.py"]

main.py

@@ -4,11 +4,9 @@ from langdetect import detect
 import torch
 import torch.nn.functional as F
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
-CACHE_DIR = os.path.join(os.getcwd(), "hf_cache")
-os.makedirs(CACHE_DIR, exist_ok=True)
-
-os.environ["HF_HOME"] = CACHE_DIR
-os.environ["TRANSFORMERS_CACHE"] = CACHE_DIR
+HF_CACHE_DIR = "/home/user/.cache/huggingface"
+os.makedirs(HF_CACHE_DIR, exist_ok=True)
+os.environ["HF_HOME"] = HF_CACHE_DIR
 
 app = Flask(__name__)