Update Dockerfile

Dockerfile

@@ -1,19 +1,29 @@
-# Use official Python slim image for a leaner base
-FROM python:3.11-slim
+# Use NVIDIA CUDA base image for GPU support
+FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04
 
 # Set working directory
 WORKDIR /app
 
-# Copy requirements file first to leverage Docker cache
-COPY requirements.txt .
-
-# Update apt-get with retries and install system dependencies
+# Install Python 3.11 and necessary tools
 RUN apt-get update --fix-missing && apt-get install -y --no-install-recommends \
+    python3.11 \
+    python3.11-dev \
+    python3-pip \
     git \
     gcc \
     build-essential \
-    libpython3.11-dev \
-    && rm -rf /var/lib/apt/lists/*
+    ffmpeg \
+    && rm -rf /var/lib/apt/lists/* \
+    && ln -sf /usr/bin/python3.11 /usr/bin/python3 \
+    && ln -sf /usr/bin/python3.11 /usr/bin/python
+
+# Set Hugging Face cache directory
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
+ENV HF_HOME=/app/.cache/huggingface
+RUN mkdir -p /app/.cache/huggingface && chmod -R 777 /app/.cache/huggingface
+
+# Copy requirements file
+COPY requirements.txt .
 
 # Update pip and install pip-tools
 RUN pip install --no-cache-dir --upgrade pip \
@@ -22,8 +32,10 @@ RUN pip install --no-cache-dir --upgrade pip \
 # Sync dependencies with pip-sync
 RUN pip-sync requirements.txt
 
-# Install Gradio and LoRA packages
-RUN pip install --no-cache-dir gradio
+# Install Gradio, diffusers, and LoRA packages
+RUN pip install --no-cache-dir gradio==4.44.0
+RUN pip install --no-cache-dir diffusers==0.30.3
+RUN pip install --no-cache-dir huggingface_hub==0.25.2
 RUN pip install git+https://github.com/cloneofsimo/lora.git
 RUN pip install git+https://github.com/microsoft/LoRA