Views
No views yet
1 from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
2 from datasets import load_dataset
3 import torch
4
5 # load model and tokenizer
6 processor = Wav2Vec2Processor.from_pretrained("Clementapa/wav2vec2-base-960h-phoneme-reco-dutch")
7 model = Wav2Vec2ForCTC.from_pretrained("Clementapa/wav2vec2-base-960h-phoneme-reco-dutch")
8
9 # load dummy dataset and read soundfiles
10 ds = load_dataset("common_voice", "nl", split="validation")
11
12 # tokenize
13 input_values = processor(ds[0]["audio"]["array"], return_tensors="pt", padding="longest").input_values # Batch size 1
14
15 # retrieve logits
16 logits = model(input_values).logits
17
18 # take argmax and decode
19 predicted_ids = torch.argmax(logits, dim=-1)
20 transcription = processor.batch_decode(predicted_ids)