Update Dockerfile

Dockerfile

@@ -1,35 +1,22 @@
-# Use Python 3.9 base image for MatAnyone compatibility
-FROM python:3.9-slim
+# ============================================================================
+# MatAnyOne + SAM2 + Gradio Dockerfile (Python 3.10 + CUDA 12.1)
+# ============================================================================
 
-# Set environment variables
+# Use Python 3.10 base
+FROM python:3.10-slim
+
+# Environment settings
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
     DEBIAN_FRONTEND=noninteractive
 
 # Install system dependencies
-RUN apt-get update && apt-get install -y \
-    git \
-    wget \
-    curl \
-    build-essential \
-    cmake \
-    pkg-config \
-    libopencv-dev \
-    python3-opencv \
-    ffmpeg \
-    libsm6 \
-    libxext6 \
-    libfontconfig1 \
-    libxrender1 \
-    libgl1-mesa-glx \
-    libglib2.0-0 \
-    libavformat-dev \
-    libavcodec-dev \
-    libavdevice-dev \
-    libavutil-dev \
-    libswscale-dev \
-    libswresample-dev \
-    libavfilter-dev \
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git wget curl build-essential cmake pkg-config \
+    ffmpeg libsm6 libxext6 libfontconfig1 libxrender1 \
+    libgl1-mesa-glx libglib2.0-0 \
+    libavformat-dev libavcodec-dev libavdevice-dev \
+    libavutil-dev libswscale-dev libswresample-dev libavfilter-dev \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
@@ -40,29 +27,28 @@ WORKDIR /app
 RUN useradd -m -u 1000 appuser && chown -R appuser:appuser /app
 USER appuser
 
-# Copy requirements first for better caching
+# Copy requirements for layer caching
 COPY --chown=appuser:appuser requirements.txt .
 
-# Install Python dependencies in stages for better error handling
+# Upgrade pip + core tools
 RUN pip install --no-cache-dir --user pip setuptools wheel
 
-# Install PyTorch first (most stable approach)
+# Install PyTorch (CUDA 12.1 build)
 RUN pip install --no-cache-dir --user \
-    torch==2.0.1 \
-    torchvision==0.15.2 \
-    torchaudio==2.0.1 \
-    --index-url https://download.pytorch.org/whl/cpu
+    torch==2.1.0 \
+    torchvision==0.16.0 \
+    torchaudio==2.1.0 \
+    --extra-index-url https://download.pytorch.org/whl/cu121
 
-# Install core dependencies
+# Install core CV/scientific dependencies
 RUN pip install --no-cache-dir --user \
-    numpy==1.24.3 \
-    opencv-python-headless==4.8.0.76 \
-    Pillow==10.0.1
-
-# Install AV package with build dependencies
-RUN pip install --no-cache-dir --user av==10.0.0
+    numpy==1.26.4 \
+    opencv-python-headless==4.10.0.84 \
+    Pillow>=10.0.1,<11.0 \
+    scipy==1.13.1 \
+    av==12.1.0
 
-# Install remaining requirements
+# Install everything else from requirements.txt
 RUN pip install --no-cache-dir --user -r requirements.txt
 
 # Add user bin to PATH
@@ -71,11 +57,11 @@ ENV PATH="/home/appuser/.local/bin:$PATH"
 # Copy application code
 COPY --chown=appuser:appuser . .
 
-# Create necessary directories
+# Create runtime directories
 RUN mkdir -p /tmp/model_cache /tmp/processing /tmp/jobs && \
     chmod 755 /tmp/model_cache /tmp/processing /tmp/jobs
 
-# Set environment variables for runtime
+# Runtime environment variables
 ENV GRADIO_SERVER_NAME="0.0.0.0" \
     GRADIO_SERVER_PORT="7860" \
     MODEL_CACHE_DIR="/tmp/model_cache" \
@@ -85,8 +71,8 @@ ENV GRADIO_SERVER_NAME="0.0.0.0" \
 HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
     CMD curl -f http://localhost:7860/ || exit 1
 
-# Expose port
+# Expose Gradio port
 EXPOSE 7860
 
-# Run the application
-CMD ["python", "app.py"]
+# Run application
+CMD ["python", "app.py"]