mirror of
https://github.com/m-bain/whisperX.git
synced 2025-07-01 18:17:27 -04:00
21 lines
622 B
Python
21 lines
622 B
Python
![]() |
import os
|
||
|
|
||
|
import pytest
|
||
|
|
||
|
import whisper
|
||
|
|
||
|
|
||
|
@pytest.mark.parametrize('model_name', whisper.available_models())
|
||
|
def test_transcribe(model_name: str):
|
||
|
model = whisper.load_model(model_name).cuda()
|
||
|
audio_path = os.path.join(os.path.dirname(__file__), "jfk.flac")
|
||
|
|
||
|
language = "en" if model_name.endswith(".en") else None
|
||
|
result = model.transcribe(audio_path, language=language, temperature=0.0)
|
||
|
assert result["language"] == "en"
|
||
|
|
||
|
transcription = result["text"].lower()
|
||
|
assert "my fellow americans" in transcription
|
||
|
assert "your country" in transcription
|
||
|
assert "do for you" in transcription
|