Update Dockerfile

Dockerfile

@@ -1,10 +1,8 @@
-# Use a slim Python base image
 FROM python:3.11-slim
 
-# Set the working directory
 WORKDIR /app
 
-# Install system dependencies
+# Install minimal system dependencies
 RUN apt-get update && apt-get install -y \
     libgomp1 \
     libglib2.0-0 \
@@ -12,26 +10,16 @@ RUN apt-get update && apt-get install -y \
     wget \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements first for better caching
+# Copy and install Python dependencies
 COPY requirements.txt .
-
-# Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Create directories for models
+# Create directory for LoRAs
 RUN mkdir -p /app/models/loras
 
-# Pre-download the PyTorch LCM model during build
-RUN python -c "from diffusers import DiffusionPipeline; \
-    pipe = DiffusionPipeline.from_pretrained('SimianLuo/LCM_Dreamshaper_v7', \
-    torch_dtype=torch.float32, safety_checker=None); \
-    pipe.to('cpu')"
-
-# Copy the application code
+# Copy the FastAPI application
 COPY app.py .
 
-# Expose the port Hugging Face Spaces uses
 EXPOSE 7860
 
-# Command to run the FastAPI application
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]