FROM anibali/pytorch:1.8.1-cuda11.1-ubuntu20.04

WORKDIR /app
COPY requirements.txt /app/requirements.txt

RUN python3 -m pip install -r requirements.txt
RUN python3 -m pip install gdown
RUN python3 -m spacy download en_core_web_sm
RUN python3 -m pip install transformers
RUN gdown --fuzzy https://drive.google.com/file/d/1_zMsEVPOUzQe8yFzRj0-Q08BA8LN4H9x/view?usp=sharing
COPY *.py /app/
EXPOSE 4999
ENTRYPOINT ["python3", "app.py"]