buzz/tests/transformers_whisper_test.py
2022-12-30 21:58:57 +00:00

11 lines
345 B
Python

from buzz.transformers_whisper import load_model
class TestTransformersWhisper:
def test_should_transcribe(self):
model = load_model('openai/whisper-tiny')
result = model.transcribe(
audio='testdata/whisper-french.mp3', language='fr', task='transcribe')
assert 'Bienvenue dans Passe' in result['text']