From b9c926456848f68e3dd784940b0b5df619adfab6 Mon Sep 17 00:00:00 2001
From: McCloudS <64094529+McCloudS@users.noreply.github.com>
Date: Mon, 5 Feb 2024 14:27:30 -0700
Subject: [PATCH] Update Dockerfile

---
 Dockerfile | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 086afb3..0fc1ac3 100755
--- 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