diff --git a/Dockerfile b/Dockerfile index 29fcc75..eb59a70 100755 --- a/Dockerfile +++ b/Dockerfile @@ -2,6 +2,8 @@ FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 WORKDIR /subgen +ADD https://raw.githubusercontent.com/McCloudS/subgen/main/requirements.txt /subgen/requirements.txt + RUN apt-get update \ && apt-get install -y \ python3 \ @@ -9,20 +11,7 @@ RUN apt-get update \ ffmpeg \ && apt-get clean \ && rm -rf /var/lib/apt/lists/* \ - && pip3 install \ - numpy \ - stable-ts \ - fastapi \ - requests \ - faster-whisper \ - uvicorn \ - python-multipart \ - python-ffmpeg \ - whisper \ - transformers \ - accelerate \ - optimum \ - watchdog + && pip3 install -r requirements.txt ENV PYTHONUNBUFFERED=1