feat: :sparkles: add docker support

.dockerignore

@@ -0,0 +1,6 @@
+venv/
+ui/__pycache__/
+outputs/
+modules/__pycache__/
+models/
+modules/yt_tmp.wav

Dockerfile

@@ -0,0 +1,22 @@
+FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04 AS runtime
+
+VOLUME [ "/Whisper-WebUI/models" ]
+VOLUME [ "/Whisper-WebUI/outputs" ]
+
+RUN apt-get update && \
+    apt-get install -y curl ffmpeg git python3 python3-pip python3-venv && \
+    rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/* && \
+    mkdir -p /Whisper-WebUI
+
+WORKDIR /Whisper-WebUI
+
+COPY . .
+
+RUN python3 -m venv venv && \
+    . venv/bin/activate && \
+    pip install --no-cache-dir -r requirements.txt
+
+ENV PATH="/Whisper-WebUI/venv/bin:$PATH"
+ENV LD_LIBRARY_PATH=/Whisper-WebUI/venv/lib64/python3.10/site-packages/nvidia/cublas/lib:/Whisper-WebUI/venv/lib64/python3.10/site-packages/nvidia/cudnn/lib
+
+ENTRYPOINT [ "python", "app.py" ]

README.md

@@ -44,6 +44,24 @@ If you have satisfied the prerequisites listed above, you are now ready to start
 
 And you can also run the project with command line arguments if you like by running `user-start-webui.bat`, see [wiki](https://github.com/jhj0517/Whisper-WebUI/wiki/Command-Line-Arguments) for a guide to arguments.
 
+## Using Docker
+
+1. build the image
+
+```sh
+docker build -t whisper-webui:latest . 
+```
+
+2. run the container
+
+```sh
+docker run --gpus all -d \
+-v /path/to/models:/Whisper-WebUI/models \
+-v /path/to/outputs:/Whisper-WebUI/outputs \
+-p 7860:7860 \
+whisper-webui:latest --server_name 0.0.0.0 --server_port 7860
+```
+
 # VRAM Usages
 This project is integrated with [faster-whisper](https://github.com/guillaumekln/faster-whisper) by default for better VRAM usage and transcription speed.