From 081c318ef4fd6949e156f65ed59fffc75047a00e Mon Sep 17 00:00:00 2001 From: McCloudS <64094529+McCloudS@users.noreply.github.com> Date: Tue, 12 Mar 2024 21:37:55 -0600 Subject: [PATCH] Update subgen.py --- subgen/subgen.py | 20 +------------------- 1 file changed, 1 insertion(+), 19 deletions(-) diff --git a/subgen/subgen.py b/subgen/subgen.py index 9de466c..f743cdd 100644 --- a/subgen/subgen.py +++ b/subgen/subgen.py @@ -367,7 +367,7 @@ def detect_language( files_to_transcribe.remove(f"Bazarr-detect-language-{random_name}") delete_model() - return {"detected_language": get_lang_pair(whisper_languages, detected_lang_code), "language_code": detected_lang_code} + return {"detected_language": whisper_languages.get(detected_lang_code, detected_lang_code) , "language_code": detected_lang_code} def start_model(): global model @@ -387,24 +387,6 @@ def delete_model(): model = None gc.collect() -def get_lang_pair(whisper_languages, key): - """Returns the other side of the pair in the Whisper languages dictionary. - - Args: - whisper_languages: A dictionary of Whisper languages. - key: The key to look up in the dictionary. - - Returns: - The other side of the pair in the Whisper languages dictionary, or None if the - key is not found in the dictionary. - """ - - other_side = whisper_languages.get(key) - if other_side is None: - return key - else: - return whisper_languages[other_side] - def gen_subtitles(file_path: str, transcribe_or_translate: str, front=True, forceLanguage=None) -> None: """Generates subtitles for a video file.