# Install dependencies
RUN apt-get update && apt-get install -y \
    python3-dev \
    build-essential \
    libjpeg-dev \
    zlib1g-dev \
    libpng-dev \
    libglib2.0-0 \
    libsm6 \
    libxrender1 \
    libxext6 \
    && rm -rf /var/lib/apt/lists/*

# Install PyTorch and TorchVision
RUN pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu

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