Update Dockerfile

Dockerfile

@@ -1,50 +1,40 @@
+# Use Python 3.10 slim as the base image for compatibility with FastSD CPU
 FROM python:3.10-slim
 
 # Set working directory
 WORKDIR /app
 
-# Install system dependencies, including libgl1 for OpenCV
+# Install system dependencies required for FastSD CPU
 RUN apt-get update && apt-get install -y \
     git \
-    git-lfs \
-    gcc \
-    python3-dev \
     libgl1 \
     libglib2.0-0 \
+    wget \
     && rm -rf /var/lib/apt/lists/*
 
-# Initialize git-lfs
-RUN git lfs install
-
-# Clone the fastsdcpu repository
+# Clone the FastSD CPU repository
 RUN git clone https://github.com/rupeshs/fastsdcpu.git .
 
-# Create writable directories for cache, configs, and results
-RUN mkdir -p /app/.cache/huggingface /app/configs /app/results && \
-    chmod -R 777 /app/.cache /app/configs /app/results
+# Create a virtual environment and activate it
+RUN python -m venv /app/env
+ENV PATH="/app/env/bin:$PATH"
 
-# Set environment variables for Hugging Face cache and OpenVINO telemetry
-ENV HF_HOME=/app/.cache/huggingface
-ENV XDG_CACHE_HOME=/app/.cache
-ENV OPENVINO_TELEMETRY=0
+# Upgrade pip and install Python dependencies
+RUN pip install --no-cache-dir --upgrade pip
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Install Python dependencies
-RUN python -m venv env && \
-    . env/bin/activate && \
-    pip install --upgrade pip && \
-    pip install torch==2.2.2 --index-url https://download.pytorch.org/whl/cpu && \
-    pip install -r requirements.txt
+# Install OpenVINO for performance optimization (optional)
+RUN pip install --no-cache-dir openvino-dev
 
-# Download a default model (corrected URL for lcm-lora-sdv1-5)
-RUN git clone https://huggingface.co/latent-consistency/lcm-lora-sdv1-5 /models/lcm-lora-sdv1-5 || \
-    echo "Model clone failed. Please check network or manually download the model."
+# Pre-download the default model (stabilityai/sd-turbo) to avoid runtime downloads
+RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='stabilityai/sd-turbo', local_dir='/app/models/sd-turbo')"
 
-# Configure the model path
-RUN echo "/models/lcm-lora-sdv1-5" > configs/lcm-lora-models.txt && \
-    chmod 666 configs/lcm-lora-models.txt
+# Expose port 8000 for the API server
+EXPOSE 8000
 
-# Expose the API port
-EXPOSE 7860
+# Set environment variables for Hugging Face Spaces
+ENV PYTHONUNBUFFERED=1
+ENV DEVICE=cpu
 
-# Activate virtual environment and run the API server
-CMD ["/bin/bash", "-c", ". env/bin/activate && python src/app.py --api"]
+# Command to run the API server
+CMD ["python", "src/app.py", "--api"]