FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
RUN apt update
RUN apt install -y build-essential
RUN pip install -U pip setuptools wheel
RUN pip install -U so-vits-svc-fork click

RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt-get install -y ffmpeg

COPY ./vocal-remover/ /vocal-remover/
ENV NUMBA_CACHE_DIR=/tmp
RUN cd /vocal-remover && pip install -r requirements.txt

ENV HF_HOME=/data/cache
ENV MPLCONFIGDIR=/data/cache

WORKDIR /
COPY ./app/ /app/

USER 1000:1000

ENTRYPOINT [ "/app/run.sh" ]