Whisper with speaker diarization
Project description
Whisper-Run
Whisper-Run is a pip CLI tool for processing audio files using Whisper models with speaker diarization capabilities. The tool allows you to process audio files, select models for audio processing, and save the results in JSON format.
It uses the OpenAI-Whisper model implementation from OpenAI Whisper, based on the ctranslate2 library from faster-whisper, and pyannote's speaker-diarization-3.1. Check their documentation if needed.
Installation
To install Whisper-Run, run the following command:
pip install whisper-run
Usage
You can call Whisper-Run from the command line using the following syntax:
whisper-run --file_path=<file_path>
Example
To process an audio file using the CPU and a specific file path:
whisper-run --device=cpu --file_path=your_file_path
When you run the command, you'll be prompted to select a model for audio processing:
[?] Select a model for audio processing:
> distil-large-v3
distil-large-v2
large-v3
large-v2
large
medium
small
base
tiny
Flags
--device
: Specify the device to use for processing (e.g.,cpu
orcuda
).--file_path
: Specify the path to the audio file you want to process.--hf_auth_token
: Optional. Pass the Hugging Face Auth Token or set theHF_AUTH_TOKEN
environment variable.
Programmatic Usage
You can also use Whisper-Run programmatically in your Python scripts. Below is a basic usage example demonstrating how to use the Whisper-Run library:
Example Script
from whisper_run import AudioProcessor
def main():
processor = AudioProcessor(file_path="your_file_path",
device="cpu",
model_name="large-v3"
)
processor.process()
if __name__ == "__main__":
main()
Contributing
Contributions are welcome! Please open an issue or submit a pull request on GitHub.
License
This project is licensed under the Apache 2.0 License.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distribution
File details
Details for the file whisper_run-1.2.5-py3-none-any.whl
.
File metadata
- Download URL: whisper_run-1.2.5-py3-none-any.whl
- Upload date:
- Size: 5.6 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.0 CPython/3.10.13
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 39e9f53a3bdefd86361b17cb03be702edb8378f22f9057b8436594bf81ddfd5c |
|
MD5 | f6a95308bf7abf9eca95c8e1482620e0 |
|
BLAKE2b-256 | e462f1afb1627f69f25c5990f90bfcdf498af0f312b8740a4ecb9b416f8ca1ab |