File size: 422 Bytes
dbc3c35 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | 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"]
|