Update Dockerfile

Dockerfile

@@ -1,42 +1,40 @@
 FROM python:3.11-slim
 
-# ---- System dependencies ----
+# System dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     libgl1 \
     libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# ---- App directory ----
 WORKDIR /app
 
-# ---- Python dependencies ----
 COPY requirements.txt /app/requirements.txt
 RUN pip install --no-cache-dir --upgrade pip
 
-# Install torch from the CPU-only wheel index first
+# CPU-only torch
 RUN pip install --no-cache-dir torch==2.3.1 --index-url https://download.pytorch.org/whl/cpu
 
-# Install the remaining Python dependencies
 RUN pip install --no-cache-dir -r /app/requirements.txt
 
-# ---- Create the non-root user ----
-RUN (useradd -m -u 1000 appuser || true) && \
-    mkdir -p /app/models /app/.cache/huggingface && \
+# Create writable cache directories BEFORE switching user
+RUN mkdir -p /app/.cache/huggingface/hub /app/models && \
     chown -R 1000:1000 /app
 
-# ---- Pre-download the model at build time ----
+# Non-root user
+RUN useradd -m -u 1000 appuser || true
+
+# Pre-download model
 ARG IMAGE_MODEL=stabilityai/sd-turbo
 ENV IMAGE_MODEL=${IMAGE_MODEL}
 ENV HF_HOME=/app/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
 ENV HF_HUB_ENABLE_HF_TRANSFER=0
 RUN python -m huggingface_hub.commands.huggingface_cli download ${IMAGE_MODEL} --local-dir /app/models/sd-turbo
 
-# ---- Copy application code ----
 COPY server.py /app/server.py
 RUN chown 1000:1000 /app/server.py
 
-# ---- Environment defaults (updated for better quality) ----
 ENV MODEL_LOCAL_DIR=/app/models/sd-turbo
 ENV DEFAULT_STEPS=4
 ENV DEFAULT_GUIDANCE=1.0