Views
No views yet
openai/whisper-large-v3-turbo1from transformers import WhisperProcessor, WhisperForConditionalGeneration
2import torch, librosa
3
4model_id = "bartelds/whisper-large-v3-turbo-fy-cv22"
5device = "cuda" if torch.cuda.is_available() else "cpu"
6
7processor = WhisperProcessor.from_pretrained(model_id)
8model = WhisperForConditionalGeneration.from_pretrained(model_id).to(device)
9
10audio, sr = librosa.load("example.wav", sr=16000, mono=True)
11inputs = processor(audio, sampling_rate=sr, return_tensors="pt")
12with torch.no_grad():
13 ids = model.generate(inputs.input_features.to(device))
14text = processor.batch_decode(ids, skip_special_tokens=True)[0]
15print(text)