Update Dockerfile

McCloudS · Feb 5, 2024 · b9c9264 · b9c9264
1 parent bfa281c
commit b9c9264
Showing 1 changed file with 17 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -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