FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime

ENV NUMBA_CACHE_DIR=/tmp

RUN apt update && apt install -y ffmpeg
RUN apt-get install -y curl git gcc libxml2-dev libxslt1-dev zlib1g-dev g++ 

RUN useradd -m -u 1000 user
USER user

WORKDIR /code

RUN chmod 777 /code

COPY . /code

RUN pip install -U pip;
RUN pip install wheel setuptools;
RUN pip install -r /code/requirements.txt

RUN git clone https://github.com/espnet/espnet.git
RUN pip install -U -e /code/espnet

EXPOSE 7860
ENV GRADIO_SERVER_NAME="0.0.0.0"

CMD ["python", "app.py"]