Update Dockerfile

Dockerfile

@@ -1,7 +1,6 @@
 FROM python:3.11-slim
 
 ENV HF_HOME=/app/hf_cache \
-    HF_HUB_ENABLE_HF_TRANSFER=1 \
     PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1
 
@@ -13,21 +12,17 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 RUN mkdir -p /app/hf_cache && chmod -R 777 /app/hf_cache
 
-RUN python - <<'PY'
+RUN python - <<'PY' || true
 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.")
+PRIMARY="cardiffnlp/twitter-roberta-base-sentiment"  # <- stable 3-class
+FALLBACK="distilbert-base-uncased-finetuned-sst-2-english"  # binary
+for mid in (PRIMARY, FALLBACK):
+    try:
+        AutoTokenizer.from_pretrained(mid)
+        AutoModelForSequenceClassification.from_pretrained(mid)
+        print("Cached:", mid)
+    except Exception as e:
+        print("Cache failed for", mid, "->", e)
 PY
 
 COPY app.py .