github/faster-whisper
1
1
Fork 0
mirror of https://github.com/SYSTRAN/faster-whisper.git synced 2026-01-09 13:38:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

change language_detection_threshold default value (#1134)

Browse Source
This commit is contained in:
Mahmoud Ashraf
2024-11-13 16:07:46 +02:00
committed by GitHub
parent fb65cd387f
commit c2bf036234
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
faster_whisper/transcribe.py
View File

@@ -692,7 +692,7 @@ class WhisperModel:
clip_timestamps: Union[str, List[float]] = "0",
hallucination_silence_threshold: Optional[float] = None,
hotwords: Optional[str] = None,
language_detection_threshold: Optional[float] = None,
language_detection_threshold: Optional[float] = 0.5,
language_detection_segments: int = 1,
) -> Tuple[Iterable[Segment], TranscriptionInfo]:
"""Transcribes an input file.
@@ -882,10 +882,7 @@ class WhisperModel:
]
# Get top language token and probability
language, language_probability = all_language_probs[0]
if (
language_detection_threshold is None
or language_probability > language_detection_threshold
):
if language_probability > language_detection_threshold:
break
detected_language_info.setdefault(language, []).append(
language_probability