Update Dockerfile

Dockerfile

@@ -1,7 +1,7 @@
 FROM python:3.11-slim
 
-ENV HF_HOME=/data/hfcache \
-    TRANSFORMERS_CACHE=/data/hfcache \
+ENV HF_HOME=/app/hf_cache \
+    HF_HUB_ENABLE_HF_TRANSFER=1 \
     PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1
 
@@ -11,6 +11,25 @@ WORKDIR /app
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+RUN mkdir -p /app/hf_cache && chmod -R 777 /app/hf_cache
+
+RUN python - <<'PY'
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+PRIMARY="cardiffnlp/twitter-roberta-base-sentiment-latest"
+FALLBACK="distilbert-base-uncased-finetuned-sst-2-english"
+print("Pre-caching:", PRIMARY)
+try:
+    AutoTokenizer.from_pretrained(PRIMARY)
+    AutoModelForSequenceClassification.from_pretrained(PRIMARY)
+    print("Primary cached.")
+except Exception as e:
+    print("Primary cache failed:", e)
+    print("Caching fallback:", FALLBACK)
+    AutoTokenizer.from_pretrained(FALLBACK)
+    AutoModelForSequenceClassification.from_pretrained(FALLBACK)
+    print("Fallback cached.")
+PY
+
 COPY app.py .
 
 EXPOSE 7860