Update Dockerfile

Dockerfile

@@ -1,7 +1,6 @@
 FROM python:3.11-slim
 
 # ---- System dependencies ----
-# Minimal apt packages needed to build/run torch + pillow on CPU.
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     libgl1 \
@@ -15,25 +14,18 @@ WORKDIR /app
 COPY requirements.txt /app/requirements.txt
 RUN pip install --no-cache-dir --upgrade pip
 
-# Install torch from the CPU-only wheel index first. This avoids pulling
-# in the ~2GB of NVIDIA CUDA/cuDNN packages that the default PyPI wheel
-# depends on, which are useless (and wasteful) on this CPU-only Space.
+# Install torch from the CPU-only wheel index first
 RUN pip install --no-cache-dir torch==2.3.1 --index-url https://download.pytorch.org/whl/cpu
 
-# Install the remaining Python dependencies.
+# 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.
+# ---- Create the non-root user ----
 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}
 ENV HF_HOME=/app/.cache/huggingface
@@ -44,10 +36,10 @@ RUN python -m huggingface_hub.commands.huggingface_cli download ${IMAGE_MODEL} -
 COPY server.py /app/server.py
 RUN chown 1000:1000 /app/server.py
 
-# ---- Environment defaults ----
+# ---- Environment defaults (updated for better quality) ----
 ENV MODEL_LOCAL_DIR=/app/models/sd-turbo
-ENV DEFAULT_STEPS=2
-ENV DEFAULT_GUIDANCE=0
+ENV DEFAULT_STEPS=4
+ENV DEFAULT_GUIDANCE=1.0
 ENV PYTHONUNBUFFERED=1
 
 USER 1000