Update Dockerfile

2024-02-10 14:30:18 -07:00
parent 5004321cb9
commit 05894c31f4
1 changed files with 5 additions and 2 deletions
--- a/7
+++ b/7
@@ -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" ]