Update Dockerfile

Dockerfile

@@ -1,4 +1,3 @@
-# Hugging Face Space (Docker) — FastAPI + sentence-transformers
 FROM python:3.11-slim
 
 # System deps
@@ -7,7 +6,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 WORKDIR /app
 
-# Python deps first (layer cache)
+# Python deps
 COPY requirements.txt /app/
 ENV PIP_NO_CACHE_DIR=1
 RUN pip install --no-cache-dir -r requirements.txt
@@ -15,13 +14,25 @@ RUN pip install --no-cache-dir -r requirements.txt
 # App
 COPY main.py /app/
 
-# Default envs (override in Space settings → Variables & Secrets)
-ENV TZ=America/New_York
+# Set a sane HOME and Hugging Face caches to a writable, persistent location
+ENV HOME=/app
 ENV MODEL_NAME=BAAI/bge-small-en-v1.5
+ENV TZ=America/New_York
+
+# Hugging Face cache locations (avoid '/.cache' permission issues)
+ENV MODEL_CACHE_DIR=/data/.cache
+ENV HF_HOME=/data/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/data/.cache/huggingface/transformers
+ENV SENTENCE_TRANSFORMERS_HOME=/data/.cache/sentence-transformers
 
-# HF Spaces exposes $PORT (usually 7860). Bind to it.
+# Ensure cache dirs exist and are writable
+RUN mkdir -p /data/.cache/huggingface/transformers /data/.cache/sentence-transformers && \
+    chmod -R 777 /data
+
+# Expose server port (Spaces sets $PORT)
 ENV PORT=7860
 EXPOSE 7860
 
-# Run the API
+# Run
 CMD ["sh", "-c", "uvicorn main:app --host 0.0.0.0 --port ${PORT} --workers 1"]
+