metadata
license: apache-2.0
datasets:
- ARTPARK-IISc/Vaani
language:
- hi
base_model:
- openai/whisper-tiny
pipeline_tag: automatic-speech-recognition
Whisper-small-vaani-hindi
This is a fine-tuned version of OpenAI's Whisper-tiny, trained on approximately 718 hours of transcribed Hindi speech from multiple datasets.
Usage
This can be used with the pipeline function from the Transformers module.
import torch
from transformers import pipeline
audio = "path to the audio file to be transcribed"
device = "cuda:0" if torch.cuda.is_available() else "cpu"
modelTags="ARTPARK-IISc/whisper-tiny-vaani-hindi"
transcribe = pipeline(task="automatic-speech-recognition", model=modelTags, chunk_length_s=30, device=device)
transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language="hi", task="transcribe")
print('Transcription: ', transcribe(audio)["text"])
Training and Evaluation
The models has finetuned using folllowing dataset Vaani ,Gramvaani IndicVoices, Fleurs,IndicTTS and Commonvoice
The performance of the model was evaluated using multiple datasets, and the evaluation results are provided below.
Dataset | WER |
---|---|
Gramvaani | 42.34 |
Fleurs | 26.39 |
IndicTTS | 11.77 |
MUCS | 39.00 |
Commonvoice | 37.95 |
Kathbath | 23.91 |
Kathbath Noisy | 29.92 |
Vaani | 33.33 |