dream

1
# syntax=docker/dockerfile:experimental
2

3
FROM pytorch/pytorch:1.5-cuda10.1-cudnn7-runtime
4

5
WORKDIR /src
6

7
ARG PRETRAINED_MODEL_NAME_OR_PATH
8
ENV PRETRAINED_MODEL_NAME_OR_PATH ${PRETRAINED_MODEL_NAME_OR_PATH}
9
ARG SERVICE_PORT
10
ENV SERVICE_PORT ${SERVICE_PORT}
11

12
ARG LANGUAGE=EN
13
ENV LANGUAGE ${LANGUAGE}
14
ARG MAX_HISTORY_DEPTH
15
ENV MAX_HISTORY_DEPTH ${MAX_HISTORY_DEPTH}
16

17
COPY ./requirements.txt /src/requirements.txt
18
RUN pip install -r /src/requirements.txt
19

20
COPY . /src
21

22
HEALTHCHECK --interval=5s --timeout=90s --retries=3 CMD curl --fail 127.0.0.1:${SERVICE_PORT}/healthcheck || exit 1
23

24

25
CMD gunicorn --workers=1 server:app -b 0.0.0.0:${SERVICE_PORT} --timeout=1200
26