whisper-base-co / README.md
Enpas's picture
Create README.md
3db093b verified
|
raw
history blame contribute delete
No virus
472 Bytes
import torch
from transformers import pipeline

device = "cuda:0" if torch.cuda.is_available() else "cpu"

transcribe = pipeline(task="automatic-speech-recognition", model="Enpas/whisper-small-co", chunk_length_s=30, device=device)
transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language="am", task="transcribe")

audio = "/content/tr_10000_tr097082.wav"
result = transcribe(audio)
print('Transcription: ', result["text"])