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

---
 Dockerfile | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 0fc1ac3..8456fcf 100755
--- 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" ]