FROM ubuntu:18.04

WORKDIR app

COPY data data
COPY model-pl2en model
COPY translate.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 python3 translate.py