| ARG ROCM_PYTORCH_IMAGE=rocm/pytorch:latest | |
| FROM ${ROCM_PYTORCH_IMAGE} | |
| WORKDIR /app | |
| RUN apt-get update \ | |
| && apt-get install -y --no-install-recommends ffmpeg git curl \ | |
| && rm -rf /var/lib/apt/lists/* | |
| COPY pyproject.toml ./ | |
| ARG INSTALL_EXTRAS=. | |
| RUN pip install --upgrade pip && pip install -e "${INSTALL_EXTRAS}" | |
| COPY app ./app | |
| EXPOSE 8000 | |
| CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000"] | |