Views
No views yet
| Training Loss | Validation | Loss Wer |
|---|---|---|
| 0.121300 | 0.103430 | 0.084904 |
1import librosa
2import torch
3from transformers import Wav2Vec2CTCTokenizer, Wav2Vec2ForCTC, Wav2Vec2Processor, TrainingArguments, Wav2Vec2FeatureExtractor, Trainer
4
5tokenizer = Wav2Vec2CTCTokenizer("./vocab.json", unk_token="[UNK]", pad_token="[PAD]", word_delimiter_token="|")
6processor = Wav2Vec2Processor.from_pretrained('boumehdi/wav2vec2-large-xlsr-moroccan-darija', tokenizer=tokenizer)
7model=Wav2Vec2ForCTC.from_pretrained('boumehdi/wav2vec2-large-xlsr-moroccan-darija')
8
9
10# load the audio data (use your own wav file here!)
11input_audio, sr = librosa.load('file.wav', sr=16000)
12
13# tokenize
14input_values = processor(input_audio, return_tensors="pt", padding=True).input_values
15
16# retrieve logits
17logits = model(input_values).logits
18
19tokens = torch.argmax(logits, axis=-1)
20
21# decode using n-gram
22transcription = tokenizer.batch_decode(tokens)
23
24# print the output
25print(transcription)