Update Dockerfile

Dockerfile

@@ -1,10 +1,11 @@
 FROM python:3.10-slim
 
+# Keep caches local to the app; make NLTK writable
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
-    HF_HOME=/root/.cache/huggingface \
-    TRANSFORMERS_CACHE=/root/.cache/huggingface/transformers \
-    PIP_NO_CACHE_DIR=1
+    HF_HOME=/app/.cache/huggingface \
+    PIP_NO_CACHE_DIR=1 \
+    NLTK_DATA=/app/nltk_data
 
 WORKDIR /app
 
@@ -18,21 +19,22 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Warm caches (NLTK + BLIP)
-RUN python - <<'PY'
-import nltk
-nltk.download('punkt'); nltk.download('averaged_perceptron_tagger')
+# Create data dirs and warm caches (NLTK + BLIP)
+RUN mkdir -p /app/nltk_data /app/data \
+ && python - <<'PY'
+import os, nltk
+print("Using NLTK_DATA:", os.environ.get("NLTK_DATA"))
+nltk.download('punkt', download_dir=os.environ['NLTK_DATA'])
+nltk.download('averaged_perceptron_tagger', download_dir=os.environ['NLTK_DATA'])
 from transformers import BlipProcessor, BlipForConditionalGeneration
 BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
 BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
-print("Models ready")
+print("Warm caches done")
 PY
 
 # App code
 COPY . .
-RUN mkdir -p /app/data
 
 EXPOSE 7860
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
-