Dockerfile

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

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

RUN python3 -m pip install --upgrade pip

RUN pip install onnxruntime-gpu

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

COPY . .

EXPOSE 8000

ENV TMP_DIRECTORY=tmp

RUN mkdir -p ${TMP_DIRECTORY}


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