transfix-translator/Dockerfile
2022-06-08 19:34:35 +01:00

23 lines
575 B
Docker

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"]