From fab9f2ea852846011270c7e74af3506427976b40 Mon Sep 17 00:00:00 2001 From: Alex Yancey Date: Wed, 5 Apr 2023 17:59:59 -0700 Subject: [PATCH] no log: Fix a bug in Whisper provider caused by undetermined language (#2110) --- libs/subliminal_patch/providers/whisperai.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/libs/subliminal_patch/providers/whisperai.py b/libs/subliminal_patch/providers/whisperai.py index 6e7df7723..0c816bb46 100644 --- a/libs/subliminal_patch/providers/whisperai.py +++ b/libs/subliminal_patch/providers/whisperai.py @@ -242,14 +242,19 @@ class WhisperAIProvider(Provider): sub = WhisperAISubtitle(language, video) sub.task = "transcribe" - if video.audio_languages: + if video.audio_languages and not (list(video.audio_languages)[0] == "und" and len(video.audio_languages) == 1): if language.alpha3 in video.audio_languages: sub.audio_language = language.alpha3 if len(list(video.audio_languages)) > 1: sub.force_audio_stream = language.alpha3 else: sub.task = "translate" - sub.audio_language = list(video.audio_languages)[0] + + eligible_languages = list(video.audio_languages) + if len(eligible_languages) > 1: + if "und" in eligible_languages: + eligible_languages.remove("und") + sub.audio_language = eligible_languages[0] else: # We must detect the language manually detected_lang = self.detect_language(video.original_path)