Update Dockerfile

Dockerfile

@@ -1,14 +1,21 @@
 FROM python:3.10-slim
 
-# Basics
+# System deps
 RUN apt-get update && apt-get install -y --no-install-recommends \
     curl ca-certificates libsndfile1 \
  && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /code
 
-# Install CPU-only torch FIRST from the CPU index to avoid CUDA deps
-# Pin a stable CPU build that plays nicely with Transformers 4.46.x
+# Make writable caches (Spaces persists /data)
+ENV TRANSFORMERS_CACHE=/data/transformers \
+    HF_HOME=/data/huggingface \
+    HUGGINGFACE_HUB_CACHE=/data/hfhub \
+    PYTHONUNBUFFERED=1
+
+RUN mkdir -p /data/transformers /data/huggingface /data/hfhub
+
+# Install CPU-only torch FIRST to avoid pulling CUDA blobs
 RUN pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cpu \
     torch==2.3.1
 
@@ -19,8 +26,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 # App
 COPY app.py .
 
-# Space variables (optional but nice for absolute URLs)
+# If you want fully-qualified URLs back, expose your space host like this:
+# (HF sets SPACE_ID automatically; this builds a FQDN inside the app)
 ENV SPACE_HOST=${SPACE_ID}.hf.space
 
-# Run
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]