FROM python:3.11-slim AS model

WORKDIR /app

RUN apt update && apt install -y libgl1-mesa-glx libglib2.0-0 && apt clean

RUN python -m pip install uv

COPY pyproject.toml .

RUN python -m uv pip install -r pyproject.toml

COPY models.py .

RUN python models.py


FROM nvcr.io/nvidia/tritonserver:24.05-py3 AS runtime

COPY --from=model --chmod=777 /app/models /models

CMD ["tritonserver", "--model-repository=/models"]