Update Dockerfile

McCloudS · Feb 10, 2024 · 05894c3 · 05894c3
1 parent 5004321
commit 05894c3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -19,9 +19,12 @@ RUN apt-get update \
     uvicorn \
     python-multipart \
     python-ffmpeg \
-    whisper
+    whisper \
+    transformers \
+    accelerate \
+    optimum
 
-ADD https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py /subgen/subgen.py
+ADD https://raw.githubusercontent.com/McCloudS/subgen/transformers/subgen/subgen.py /subgen/subgen.py
 
 CMD [ "python3", "-u", "./subgen.py" ]