Update Dockerfile

Dockerfile

@@ -7,31 +7,43 @@ WORKDIR /app
 # 3. Copy all your project files (app.py, CatVTON folder, etc.) 
 COPY . .
 
-# 4. Install system dependencies needed for 'av' and 'opencv'
 # 4. Install system dependencies needed for 'av' and 'opencv'
 RUN apt-get update && apt-get install -y \
     ffmpeg \
     libgl1 \
     && apt-get clean
 
-# 5. Run your EXACT Kaggle installation script
-# This gives you full control over the order.
-# We use --mount for better pip caching during builds
+# 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 && \
-    pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118 && \
+    pip install matplotlib==3.7.2
+
+# --- Step 5b: Install PyTorch stack
+RUN --mount=type=cache,target=/root/.cache/pip \
+    pip install torch==2.1.0 torchvision==0.16.0 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 && \
+    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 && \
+    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