Dockerfile

FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 

WORKDIR /app
ENV DEBIAN_FRONTEND=noninteractive

RUN apt-get update && apt-get install -y \
    ffmpeg \
    tk \
    python3-tk \
    python3-pip \
    && rm -rf /var/lib/apt/lists/*

RUN --mount=type=cache,target=/root/.cache \
    pip3 install "audio-separator[gpu]" fastapi uvicorn python-multipart ffmpeg-python pydantic youtube-transcript-api

COPY . /app

EXPOSE 7860

RUN mkdir -p ${TMP_DIRECTORY}
ENV NUMBA_CACHE_DIR=/tmp

CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]