Update Dockerfile

Dockerfile

@@ -1,62 +1,87 @@
-# 1. Use a standard Python 3.10 image (matches HF Spaces)
-FROM python:3.10-slim
+# Use CUDA-enabled base image for GPU support
+FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
-# 2. Set the working directory inside the container
-WORKDIR /app
+# Set environment variables
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED=1
+ENV CUDA_HOME=/usr/local/cuda-11.8
 
-# 3. Copy all your project files (app.py, CatVTON folder, etc.) 
-COPY . .
+WORKDIR /app
 
-# 4. Install system dependencies needed for 'av' and 'opencv'
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
-    ffmpeg \
-    libgl1 \
-    && apt-get clean
-
-# 5. Run your Kaggle installation script in smaller, logical steps
-#    This prevents the "Out of Memory" (Exit 137) error.
-
-# --- Step 5a: Uninstall and install core scientific stack
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip uninstall -y numpy scipy scikit-learn matplotlib huggingface-hub accelerate peft && \
-    pip install numpy==1.26.4 && \
-    pip install scipy==1.11.4 && \
-    pip install scikit-learn==1.3.2 && \
-    pip install matplotlib==3.7.2
-
-# --- Step 5b: Install PyTorch stack (one-by-one to save RAM)
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install torch==2.1.0 --index-url https://download.pytorch.org/whl/cu118
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
-
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
-
-# --- Step 5c: Install Hugging Face stack
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install huggingface_hub==0.20.0 && \
-    pip install safetensors==0.4.1 && \
-    pip install diffusers==0.25.0 && \
-    pip install transformers==4.36.0 && \
-    pip install accelerate==0.28.0 && \
-    pip install peft==0.8.2
-
-# --- Step 5d: Install Vision/Utility stack
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install opencv-python-headless==4.8.1.78 pillow==11.3.0 && \
-    pip install fvcore==0.1.5.post20221221 iopath==0.1.10 av==16.0.1 omegaconf==2.3.0
-
-# --- Step 5e: Install Gradio stack
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install ipywidgets==8.1.5 && \
-    pip install gradio==4.31.0
-
-# 6. CRITICAL: Re-run your "Step 7" to lock the hub version
-# This runs as its own step to guarantee it wins any conflicts
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install --force-reinstall huggingface_hub==0.20.0 --no-deps
-
-# 7. Tell the container what command to run to start your Gradio app
-CMD ["gradio", "app.py"]
+    python3.10 \
+    python3-pip \
+    git \
+    wget \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    libgomp1 \
+    libgl1-mesa-glx \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create symbolic links for python
+RUN ln -sf /usr/bin/python3.10 /usr/bin/python && \
+    ln -sf /usr/bin/pip3 /usr/bin/pip
+
+# Upgrade pip
+RUN pip install --upgrade pip setuptools wheel
+
+# ===== CRITICAL: Install packages in EXACT ORDER to maintain compatibility =====
+
+# Step 1: Install NumPy 1.26.4 FIRST (LOCKED)
+RUN pip install --no-cache-dir numpy==1.26.4
+
+# Step 2: Install compatible scientific packages
+RUN pip install --no-cache-dir \
+    scipy==1.11.4 \
+    scikit-learn==1.3.2 \
+    matplotlib==3.7.2
+
+# Step 3: Install PyTorch with CUDA 11.8 support
+RUN pip install --no-cache-dir \
+    torch==2.1.0 \
+    torchvision==0.16.0 \
+    torchaudio==2.1.0 \
+    --index-url https://download.pytorch.org/whl/cu118
+
+# Step 4: Install AI/ML packages with LOCKED versions
+RUN pip install --no-cache-dir \
+    huggingface_hub==0.20.0 \
+    safetensors==0.4.1 \
+    diffusers==0.25.0 \
+    transformers==4.36.0 \
+    accelerate==0.28.0 \
+    peft==0.8.2
+
+# Step 5: Install computer vision and utility packages
+RUN pip install --no-cache-dir \
+    opencv-python-headless==4.8.1.78 \
+    pillow
+
+# Step 6: Install additional dependencies
+RUN pip install --no-cache-dir \
+    fvcore \
+    iopath \
+    av \
+    omegaconf
+
+# Step 7: Install Gradio for web interface (Hugging Face Spaces)
+RUN pip install --no-cache-dir gradio
+
+# Step 8: Force reinstall huggingface_hub to lock version (prevent auto-upgrades)
+RUN pip install --force-reinstall --no-cache-dir huggingface_hub==0.20.0 --no-deps
+
+# Copy application code
+COPY . /app
+
+# Create necessary directories
+RUN mkdir -p /app/models /app/outputs
+
+# Expose port for Hugging Face Spaces (default: 7860)
+EXPOSE 7860
+
+# Set the entrypoint
+CMD ["python", "app.py"]