Dockerfile

FROM python:3.9-slim

# Install system dependencies
RUN apt-get update && apt-get install -y \
    git \
    gcc \
    libglib2.0-0 \
    libsm6 \
    libxext6 \
    libxrender-dev \
    ffmpeg \
    && rm -rf /var/lib/apt/lists/*

# Install PyTorch CPU version
RUN pip install torch==2.0.1+cpu torchvision==0.15.2+cpu torchaudio==2.0.2+cpu --index-url https://download.pytorch.org/whl/cpu

# Install required Python packages
RUN pip install opencv-python Pillow gdown gradio==4.25.0 pycocotools scikit-image shapely

# Install detectron2 manually
RUN pip install 'git+https://github.com/facebookresearch/detectron2.git'

# Set Matplotlib temp folder
ENV MPLCONFIGDIR=/tmp/matplotlib

# Copy code
COPY . /app
WORKDIR /app

# Run Gradio app
CMD ["python3", "app.py"]