Update Dockerfile

This commit is contained in:
McCloudS
2024-02-05 14:27:30 -07:00
committed by GitHub
parent bfa281c1c4
commit b9c9264568

View File

@@ -3,10 +3,23 @@ FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
VOLUME /subgen
WORKDIR /subgen
RUN apt-get update && apt-get -y install python3 python3-pip ffmpeg
RUN pip3 install numpy stable-ts fastapi requests faster-whisper uvicorn python-multipart python-ffmpeg whisper
ENV PYTHONUNBUFFERED 1
RUN apt-get update \
&& apt-get install -y \
python3 \
python3-pip \
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
ADD https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py /subgen/subgen.py