From nvcr.io/nvidia/pytorch:19.04-py3

RUN apt-get -y update

RUN apt-get install -y python3-pip software-properties-common wget ffmpeg

RUN apt-get -y update

RUN mkdir -p /root/.torch/models

RUN mkdir -p /data/models

ADD . /data/

WORKDIR /data

RUN pip install -r requirements.txt

RUN pip install  Flask

RUN pip install Pillow

RUN pip install scikit-image

RUN pip install requests

EXPOSE 5000

ENTRYPOINT ["python3"]

CMD ["app.py"]