Update Dockerfile

Dockerfile

@@ -1,32 +1,28 @@
-# Use a lightweight Python image
 FROM python:3.10-slim
 
 # Install system dependencies
-# Note: libgl1 is the updated replacement for libgl1-mesa-glx
 RUN apt-get update && apt-get install -y \
     libgl1 \
     libglib2.0-0 \
     git \
-    python3-pip \
     && rm -rf /var/lib/apt/lists/*
 
-# Set working directory
 WORKDIR /app
 
-# Clone the FastSD CPU repository
+# 1. Force install CPU-only version of Torch first 
+# This prevents pip from trying to download the 5GB+ NVIDIA drivers
+RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu
+
+# 2. Clone the repo
 RUN git clone https://github.com/rupeshs/fastsdcpu.git .
 
-# Install dependencies 
-# We add 'fastapi' and 'uvicorn' to ensure the web server layers work well in a container
+# 3. Install requirements but skip the heavy torch/cuda resolution
+# We use --no-deps or specific constraints to keep the build fast
 RUN pip install --no-cache-dir -r requirements.txt
 RUN pip install --no-cache-dir openvino-dev fastapi uvicorn
 
-# Set environment variable to skip the opening of a browser window (prevents crashes in Docker)
-ENV SD_WEBUI_RELOAD=1
-
 # Hugging Face Spaces run on port 7860
 EXPOSE 7860
 
 # Command to start the web UI
-# Added --browser false to ensure it doesn't try to launch a window inside the container
 CMD ["python3", "src/app.py", "--host", "0.0.0.0", "--port", "7860"]