Update Dockerfile

Dockerfile

@@ -1,10 +1,10 @@
-# Use a slim Python base image for efficiency
+# Use a slim Python base image
 FROM python:3.11-slim
 
 # Set the working directory
 WORKDIR /app
 
-# Install system dependencies including git for cloning the model repo
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     libgomp1 \
     libglib2.0-0 \
@@ -12,20 +12,20 @@ RUN apt-get update && apt-get install -y \
     wget \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy the requirements file first for better caching
+# Copy requirements first for better caching
 COPY requirements.txt .
 
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Create a directory for models and clone the OpenVINO model from Hugging Face
-RUN mkdir -p /app/models && \
-    git clone https://huggingface.co/rupeshs/LCM-dreamshaper-v7-openvino /app/models/LCM-dreamshaper-v7-openvino && \
-    # Create a dedicated directory for LoRAs
-    mkdir -p /app/models/loras
+# Create directories for models
+RUN mkdir -p /app/models/loras
 
-# Set environment variable to point to the local model path
-ENV OV_MODEL_PATH="/app/models/LCM-dreamshaper-v7-openvino"
+# 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 app.py .