This repository uses ffmpeg
and OpenAI's Whisper to automatically generate and overlay subtitles on any video.
python3 -m venv myenv
#For Linux
source myenv/bin/activate
#for Windows
myenv\Scripts\activate
To get started, you'll need Python 3.7 or newer. Install the binary by running the following command:
pip install git+https://github.com/m1guelpf/auto-subtitle.git
pip install ffmpeg-python
You'll also need to install ffmpeg
, which is available from most package managers:
# on Ubuntu or Debian
sudo apt update && sudo apt install ffmpeg
# on MacOS using Homebrew (https://brew.sh/)
brew install ffmpeg
# on Windows using Chocolatey (https://chocolatey.org/)
choco install ffmpeg
The following command will generate a subtitled/video.mp4
file contained the input video with overlayed subtitles.
auto_subtitle /path/to/video.mp4 -o subtitled/
The default setting (which selects the small
model) works well for transcribing English. You can optionally use a bigger model for better results (especially with other languages). The available models are tiny
, tiny.en
, base
, base.en
, small
, small.en
, medium
, medium.en
, large
.
auto_subtitle /path/to/video.mp4 --model medium
Adding --task translate
will translate the subtitles into English:
auto_subtitle /path/to/video.mp4 --task translate
Run the following to view all available options:
auto_subtitle --help
wget https://raw.githubusercontent.com/Nishan666/auto-subtitle/main/bulk.py
python3 bulk.py <video_folder_path> <model>
This script is open-source and licensed under the MIT License. For more details, check the LICENSE file.