FROM ubuntu:18.04

WORKDIR app

COPY in in
COPY out out
COPY model-pl2en model-pl2en
COPY translate.py .
COPY resources resources
COPY app.py .

RUN apt-get update \
  && apt-get install -y python3-pip python3-dev \
  && cd /usr/local/bin \
  && ln -s /usr/bin/python3 python \
  && pip3 install --upgrade pip

RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
RUN pip3 install protobuf==3.17.3 transformers nltk sentencepiece
RUN pip3 install flask_restful werkzeug

CMD ["python3", "-m", "flask", "run", "--host=0.0.0.0"]