Update Dockerfile

Dockerfile

@@ -23,28 +23,32 @@ RUN pip install --no-cache-dir torch==2.3.1 --index-url https://download.pytorch
 # Install the remaining Python dependencies.
 RUN pip install --no-cache-dir -r /app/requirements.txt
 
+# ---- Create the non-root user required by HF Spaces before downloading ----
+# so the downloaded model is owned by the right user from the start
+# (avoids a slow recursive chown of multi-GB model weights later).
+# "|| true" guards against a base-image UID/GID 1000 already existing.
+RUN (useradd -m -u 1000 appuser || true) && \
+    mkdir -p /app/models /app/.cache/huggingface && \
+    chown -R 1000:1000 /app
+
 # ---- Pre-download the model at build time ----
 # This bakes the model weights into the image so the container starts
 # fast and does not need network access to Hugging Face at runtime.
 ARG IMAGE_MODEL=stabilityai/sd-turbo
 ENV IMAGE_MODEL=${IMAGE_MODEL}
-RUN hf download ${IMAGE_MODEL} --local-dir /app/models/sd-turbo
+ENV HF_HOME=/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 ----
 ENV MODEL_LOCAL_DIR=/app/models/sd-turbo
 ENV DEFAULT_STEPS=2
 ENV DEFAULT_GUIDANCE=0
 ENV PYTHONUNBUFFERED=1
-ENV HF_HOME=/app/.cache/huggingface
-ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
-
-# ---- HF Spaces requires a non-root user with UID 1000 ----
-RUN useradd -m -u 1000 appuser && \
-    mkdir -p /app/.cache/huggingface && \
-    chown -R appuser:appuser /app
 
 USER 1000