FROM nvidia/cuda:12.1.1-base-ubuntu22.04

ENV TZ=Pacific/Auckland
ENV DEBIAN_FRONTEND=noninteractive

RUN apt-get update 
RUN apt-get update && apt-get install -y tzdata git pandoc ffmpeg python3 python3-pip

ADD ./requirements.txt /srv/requirements.txt
RUN pip install -r /srv/requirements.txt

RUN pip install --no-cache-dir         datasets         "huggingface-hub>=0.12.1" "protobuf<4" "click<8.1"

ADD . /srv

WORKDIR /srv 
CMD ["gradio", "app.py"]