docker #22

.dockerignore

@@ -0,0 +1,23 @@
+__pycache__
+*.pyc
+*.pyo
+*.pyd
+.Python
+env/
+venv/
+pip-log.txt
+pip-delete-this-directory.txt
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.log
+.git
+.gitignore
+.mypy_cache
+.pytest_cache
+.hypothesis
+.idea

Dockerfile

@@ -0,0 +1,24 @@
+#FROM python:3.10.6
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+
+WORKDIR /subsai
+
+COPY requirements.txt .
+
+ENV DEBIAN_FRONTEND noninteractive
+RUN apt-get update && \
+    apt-get upgrade -y && \
+    apt-get install -y git && \
+    apt-get -y install gcc mono-mcs && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY pyproject.toml .
+COPY ./src ./src
+COPY ./assets ./assets
+
+RUN pip install .
+
+CMD ["python", "src/subsai/webui.py"]
+

README.md

@@ -96,16 +96,18 @@ _Quoted from the official openai/whisper installation_
 ```shell
 pip install git+https://github.com/abdeladim-s/subsai
 ```
+
 # Usage
-### Web UI
+### Web-UI
 
-To use the web UI, run the following command on the terminal
+To use the web-UI, run the following command on the terminal
 ```shell
 subsai-webui
 ```
-
 And a web page will open on your default browser, otherwise navigate to the links provided by the command
 
+You can also run the Web-UI using [Docker](#Docker).
+
 ### CLI
 
 ```shell
@@ -176,6 +178,14 @@ Simple examples can be found in the [examples](https://github.com/abdeladim-s/su
   <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
 </a>
 
+# Docker
+If you only want to use the Web-UI, you can use docker as follows:
+1. Make sure that you have `docker` installed.
+2. Clone and cd to the repository
+3. ```docker compose build```
+4. ```docker compose run -v /path/to/your/media_files/folder:/media_files subsai-webui```
+5. You can access your media files through the mounted `media_files` folder.
+
 # Notes
 * If you have an NVIDIA graphics card, you may need to install [cuda](https://docs.nvidia.com/cuda/#installation-guides) to use the GPU capabilities.
 * Transcription time is shown on the terminal, keep an eye on it while running the web UI. 

docker-compose.yml

@@ -0,0 +1,7 @@
+version: '3.8'
+
+services:
+  subsai-webui:
+    build:
+      context: .
+      dockerfile: Dockerfile

requirements.txt

@@ -1,13 +1,13 @@
-torch
+torch==2.0.1
 openai-whisper==20230124
 streamlit~=1.20.0
 streamlit_player~=0.1.5
 streamlit-aggrid~=0.3.4
 ffsubsync~=0.4.24
 whisper-timestamped @ git+https://github.com/linto-ai/whisper-timestamped
-pandas
+pandas==1.5.3
 pysubs2~=1.6.0
 pywhispercpp==1.1.1
-dl_translate
-faster_whisper
+dl_translate==0.2.6
+faster_whisper==0.5.1
 whisperx @ git+https://github.com/m-bain/whisperx.git