Dockerfile

FROM python:3.10-slim

WORKDIR /app
ENV PIP_NO_CACHE_DIR=1

COPY requirements.txt /app/requirements.txt

RUN apt-get update && apt-get install -y \
    git build-essential ffmpeg libsm6 libxext6 curl \
    && rm -rf /var/lib/apt/lists/*

RUN pip install --upgrade pip && pip install -r requirements.txt

# Detectron2 CPU build (from source) - may take time on first build
RUN pip install 'git+https://github.com/facebookresearch/detectron2.git@v0.5#egg=detectron2'

COPY app.py /app/app.py
COPY inference.py /app/inference.py
COPY model_final.pth /app/model_final.pth

ENV USE_GPU=false
EXPOSE 7860
CMD ["python", "app.py"]