Update Dockerfile

Dockerfile

@@ -1,20 +1,17 @@
-FROM python:3.10
+FROM python:3.10-slim
 
-# Install system dependencies
-RUN apt-get update && apt-get install -y git
-
-# Set workdir
 WORKDIR /app
 
-# Copy files
+RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*
+
 COPY app.py /app/app.py
 COPY static /app/static
+COPY requirements.txt /app/requirements.txt
 
-# Install Python dependencies
-RUN pip install --no-cache-dir fastapi uvicorn diffusers transformers accelerate safetensors torch Pillow
+# Install CPU-only torch wheel (this will install CPU PyTorch)
+RUN pip install --no-cache-dir torch --extra-index-url https://download.pytorch.org/whl/cpu
+RUN pip install --no-cache-dir -r /app/requirements.txt
 
-# Expose port
 EXPOSE 7860
 
-# Run app
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]