Update Dockerfile

Dockerfile

@@ -1,23 +1,39 @@
+# ---- Base image
 FROM python:3.11-slim
 
+# ---- Env
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
     HF_HOME=/root/.cache/huggingface \
-    TRANSFORMERS_CACHE=/root/.cache/huggingface/transformers
+    TRANSFORMERS_CACHE=/root/.cache/huggingface/transformers \
+    HF_HUB_ENABLE_HF_TRANSFER=1 \
+    PIP_NO_CACHE_DIR=1
 
 WORKDIR /app
 
-# system basics (tiny)
-RUN apt-get update && apt-get install -y --no-install-recommends git && \
-    rm -rf /var/lib/apt/lists/*
+# ---- Minimal OS deps (no OpenGL libs needed)
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git git-lfs \
+ && rm -rf /var/lib/apt/lists/* \
+ && git lfs install
 
+# ---- Python deps
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# pre-download tiny NLTK bits so first request is warm
-RUN python -c "import nltk; nltk.download('punkt'); nltk.download('averaged_perceptron_tagger')"
+# ---- Warm caches (NLTK + BLIP weights) so first request is fast
+RUN python - <<'PY'
+import nltk
+nltk.download('punkt'); nltk.download('averaged_perceptron_tagger')
+from transformers import BlipForConditionalGeneration, BlipProcessor
+BlipProcessor.from_pretrained('Salesforce/blip-image-captioning-base')
+BlipForConditionalGeneration.from_pretrained('Salesforce/blip-image-captioning-base')
+print("Models ready")
+PY
 
+# ---- App code
 COPY . .
+RUN mkdir -p /app/data
 
 EXPOSE 7860
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]