Update Dockerfile

Dockerfile

@@ -1,47 +1,49 @@
-FROM python:3.11-slim
+FROM python:3.12-slim
 
-# 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/*
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED=1
+ENV PIP_NO_CACHE_DIR=1
 
 WORKDIR /app
 
-COPY requirements.txt /app/requirements.txt
-RUN pip install --no-cache-dir --upgrade pip
-
-# CPU-only torch
-RUN pip install --no-cache-dir torch==2.3.1 --index-url https://download.pytorch.org/whl/cpu
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git \
+    cmake \
+    build-essential \
+    curl \
+    ca-certificates \
+    libgomp1 \
+    && rm -rf /var/lib/apt/lists/*
 
-RUN pip install --no-cache-dir -r /app/requirements.txt
 
-# Create writable cache directories BEFORE switching user
-RUN mkdir -p /app/.cache/huggingface/hub /app/models && \
-    chown -R 1000:1000 /app
+# Build llama.cpp for CPU.
+RUN git clone --depth 1 https://github.com/ggml-org/llama.cpp.git /tmp/llama.cpp \
+    && cmake -S /tmp/llama.cpp -B /tmp/llama.cpp/build \
+        -DGGML_NATIVE=OFF \
+        -DGGML_OPENMP=ON \
+        -DLLAMA_BUILD_SERVER=ON \
+        -DLLAMA_BUILD_TESTS=OFF \
+        -DLLAMA_BUILD_EXAMPLES=ON \
+    && cmake --build /tmp/llama.cpp/build \
+        --config Release \
+        --target llama-server \
+        -j2 \
+    && mkdir -p /opt/llama \
+    && cp /tmp/llama.cpp/build/bin/llama-server /opt/llama/llama-server \
+    && cp /tmp/llama.cpp/build/bin/*.so* /opt/llama/ 2>/dev/null || true \
+    && rm -rf /tmp/llama.cpp
 
-# 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
+# Hugging Face downloader.
+RUN pip install --upgrade pip \
+    && pip install huggingface_hub
 
-COPY server.py /app/server.py
-RUN chown 1000:1000 /app/server.py
 
-ENV MODEL_LOCAL_DIR=/app/models/sd-turbo
-ENV DEFAULT_STEPS=4
-ENV DEFAULT_GUIDANCE=1.0
-ENV PYTHONUNBUFFERED=1
+COPY start.sh /app/start.sh
+RUN chmod +x /app/start.sh
 
-USER 1000
 
 EXPOSE 7860
 
-CMD ["python", "server.py"]
+
+CMD ["/app/start.sh"]