clean Dockerfile

This commit is contained in:
ale 2021-07-30 17:52:29 +02:00
parent b650dc95df
commit 0cb722d1ce

View File

@ -1,11 +1,10 @@
FROM python:3
RUN curl https://sh.rustup.rs -sSf | sh -s -- -y
RUN . /root/.cargo/env && pip3 install transformers torch flask
RUN . /root/.cargo/env && pip3 install transformers torch
RUN mkdir -p /maria
WORKDIR /maria
RUN git clone https://huggingface.co/BSC-TeMU/roberta-base-bne
COPY ./roberta.py /maria/roberta.py
RUN python3 roberta.py "hola <mask>"
CMD "hola <mask>"
ENTRYPOINT ["python3", "roberta.py"]