<filename>hub/executors/encoders/nlp/transformers-hitscir/Dockerfile
FROM pytorch/pytorch:latest

RUN pip install transformers --no-cache-dir --compile && \
    python -c "from transformers import *; x='hfl/chinese-roberta-wwm-ext'; BertModel.from_pretrained(x); BertTokenizer.from_pretrained(x)"

# jina is not public yet, so we add from the local host and install it
ADD jina .src

RUN pip install .src/.

ENV TMP_WORKSPACE /tmp/jina/webqa/
ENV BATCH_SIZE 8

ADD *.py *.yml ./

ENTRYPOINT ["jina", "pod", "--yaml-path", "roberta.yml"]