Create Dockerfile

Dockerfile

@@ -0,0 +1,46 @@
+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 \
+    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 && \
+    pip install --no-cache-dir -r /app/requirements.txt
+
+# ---- 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
+
+# ---- Copy application code ----
+COPY server.py /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
+
+EXPOSE 7860
+
+CMD ["python", "server.py"]