FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y libncurses5 python3 python3-pip curl git apt-utils
RUN pip3 install torch==1.8.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html
RUN pip3 install transformers pika
RUN pip3 install prometheus_client

EXPOSE 8080
EXPOSE 80
EXPOSE 4369
EXPOSE 5672
EXPOSE 25672
EXPOSE 15672
EXPOSE 15692

WORKDIR /usr/src/app
COPY . .
CMD [ "python3", "Models/api_serve.py" ]