pbr-private/Dockerfile
2022-05-29 18:32:42 +01:00

18 lines
449 B
Docker

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