New PR for Faster Whisper: Batching Support, Speed Boosts, and Qualit…

…y Enhancements (SYSTRAN#856) Batching Support, Speed Boosts, and Quality Enhancements --------- Co-authored-by: Hargun Mujral <[email protected]> Co-authored-by: MahmoudAshraf97 <[email protected]>
mobiusml · Oct 2, 2024 · 5b2cc66 · 5b2cc66
1 parent 1c22ff3
commit 5b2cc66
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_transcribe.py b/tests/test_transcribe.py
@@ -145,6 +145,13 @@ def test_stereo_diarization(data_dir):
     assert transcription == "The horizon seems extremely distant."
 
 
+def test_multisegment_lang_id(physcisworks_path):
+    model = WhisperModel("tiny")
+    language_info = model.detect_language_multi_segment(physcisworks_path)
+    assert language_info["language_code"] == "en"
+    assert language_info["language_confidence"] > 0.8
+
+
 def test_suppressed_tokens_minus_1():
     model = WhisperModel("tiny.en")