Update Dockerfile

This commit is contained in:
McCloudS
2024-03-19 14:40:50 -06:00
committed by GitHub
parent a97cec3c29
commit fe046af2aa

View File

@@ -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