Ravi Theja fcf9e87f90
Add transcription using gladia (#164)
Co-authored-by: Jerry Liu <jerryjliu98@gmail.com>
2023-04-07 22:58:04 -07:00

32 lines
1.2 KiB
Markdown

# Audio File Loader
This loader uses Gladia's OpenAI's Whisper model to transcribe the text of an audio file or the audio track of a video file. The file formats .mp3 and .mp4 are preferred. A single local file is passed in each time you call `load_data`.
## Usage
To use this loader, you need to pass in a `Path` to a local file.
Please check following parameters on [Gladia](https://www.gladia.io/) before proceeding further.
1. gladia_api_key
2. diarization_max_speakers
3. language
4. language_behaviour
5. target_translation_language
6. transcription_hint
You need to signup on [Gladia](https://www.gladia.io/) to get `API-KEY`
```python
from pathlib import Path
from llama_index import download_loader
AudioTranscriber = download_loader("AudioTranscriber")
# using gladia
loader = AudioTranscriber(model_type = 'gladia', gladia_api_key = 'YOUR API KEY')
documents = loader.load_data(file=Path('./podcast.mp3'))
```
This loader is designed to be used as a way to load data into [LlamaIndex](https://github.com/jerryjliu/gpt_index/tree/main/gpt_index) and/or subsequently used as a Tool in a [LangChain](https://github.com/hwchase17/langchain) Agent. See [here](https://github.com/emptycrown/llama-hub/tree/main) for examples.