/auto-subtitle

Automatically generate and overlay subtitles for any video.

Primary LanguagePythonMIT LicenseMIT

Automatic subtitles in your videos

This fork has Windows path fix and ffmpeg use NVENC. Tested fully working on Windows 11 & RTX 2080

This repository uses ffmpeg and OpenAI's Whisper to automatically generate and overlay subtitles on any video.

Installation

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/brokeDude2901/auto-subtitle.git
pip install python-ffmpeg

Usage

The following command will generate a subtitled/video.mp4 file contained the input video with overlayed subtitles.

python -m auto_subtitle.cli /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.

python -m auto_subtitle.cli /path/to/video.mp4 --model medium

Adding --task translate will translate the subtitles into English:

python -m auto_subtitle.cli /path/to/video.mp4 --task translate

Run the following to view all available options:

python -m auto_subtitle.cli --help

License

This script is open-source and licensed under the MIT License. For more details, check the LICENSE file.