Update Dockerfile

Dockerfile

@@ -1,43 +1,45 @@
-# Use an Ubuntu-based image with Python 3.10
-FROM python:3.10-slim
-
-# Set working directory
-WORKDIR /app
-
-# Install system dependencies
-RUN apt-get update && apt-get install -y \
-    ffmpeg \
-    libsm6 \
-    libxext6 \
-    && rm -rf /var/lib/apt/lists/*
-
-# Install UV for faster dependency installation
-RUN pip install uv
-
-# Copy requirements and install
-COPY requirements.txt .
-RUN uv pip install --system -r requirements.txt
-
-# Create cache directories with write permissions
-RUN mkdir -p /app/cache/huggingface /app/cache/openvino /app/matplotlib_cache /app/openvino_cache \
-    && chmod -R 777 /app/cache /app/matplotlib_cache /app/openvino_cache
-
-# Set environment variables for cache directories
-ENV HF_HOME=/app/cache/huggingface
-ENV MPLCONFIGDIR=/app/matplotlib_cache
-ENV OPENVINO_TELEMETRY_DIR=/app/openvino_cache
-
-# Pre-download base SDXL model
-RUN python -c "from optimum.intel.openvino import OVStableDiffusionPipeline; \
-    OVStableDiffusionPipeline.from_pretrained('rupeshs/hyper-sd-sdxl-1-step-openvino-int8', ov_config={'CACHE_DIR': '/app/cache/openvino'})"
-
-
-# Copy application code
-COPY app.py .
-
-# Expose port (default 5000, configurable via PORT env variable)
-ENV PORT=7860
-EXPOSE $PORT
-
-# Command to run the Flask app
+# Use an Ubuntu-based image with Python 3.10
+FROM python:3.10-slim
+
+# Set working directory
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install UV for faster dependency installation
+RUN pip install uv
+
+# Copy requirements and install
+COPY requirements.txt .
+RUN uv pip install --system -r requirements.txt
+
+RUN pip install -q "optimum-intel[openvino,diffusers]@git+https://github.com/huggingface/optimum-intel.git" "transformers>=4.33" --extra-index-url https://download.pytorch.org/whl/cpu
+
+# Create cache directories with write permissions
+RUN mkdir -p /app/cache/huggingface /app/cache/openvino /app/matplotlib_cache /app/openvino_cache \
+    && chmod -R 777 /app/cache /app/matplotlib_cache /app/openvino_cache
+
+# Set environment variables for cache directories
+ENV HF_HOME=/app/cache/huggingface
+ENV MPLCONFIGDIR=/app/matplotlib_cache
+ENV OPENVINO_TELEMETRY_DIR=/app/openvino_cache
+
+# Pre-download base SDXL model
+RUN python -c "from optimum.intel.openvino import OVStableDiffusionPipeline; \
+    OVStableDiffusionPipeline.from_pretrained('rupeshs/hyper-sd-sdxl-1-step-openvino-int8', ov_config={'CACHE_DIR': '/app/cache/openvino'})"
+
+
+# Copy application code
+COPY app.py .
+
+# Expose port (default 5000, configurable via PORT env variable)
+ENV PORT=7860
+EXPOSE $PORT
+
+# Command to run the Flask app
 CMD ["python", "app.py"]