diff --git a/subgen/subgen.py b/subgen/subgen.py index 4d2fd91..f13b504 100644 --- a/subgen/subgen.py +++ b/subgen/subgen.py @@ -316,7 +316,7 @@ def asr( start_time = time.time() start_model() - files_to_transcribe.insert(0, f"Bazarr-detect-langauge-{random_name}") + files_to_transcribe.insert(0, f"Bazarr-detect-language-{random_name}") if(hf_transformers): result = model.transcribe(np.frombuffer(audio_file.file.read(), np.int16).flatten().astype(np.float32) / 32768.0, task=task, input_sr=16000, language=language, batch_size=hf_batch_size, progress_callback=progress) else: @@ -328,8 +328,8 @@ def asr( except Exception as e: logging.info(f"Error processing or transcribing Bazarr {audio_file.filename}: {e}") finally: - if f"Bazarr-detect-langauge-{random_name}" in files_to_transcribe: - files_to_transcribe.remove(f"Bazarr-detect-langauge-{random_name}") + if f"Bazarr-detect-language-{random_name}" in files_to_transcribe: + files_to_transcribe.remove(f"Bazarr-detect-language-{random_name}") delete_model() if result: return StreamingResponse( @@ -352,7 +352,7 @@ def detect_language( random_name = random.choices("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890", k=6) start_model() - files_to_transcribe.insert(0, f"Bazarr-detect-langauge-{random_name}") + files_to_transcribe.insert(0, f"Bazarr-detect-language-{random_name}") if(hf_transformers): detected_lang_code = model.transcribe(whisper.pad_or_trim(np.frombuffer(audio_file.file.read(), np.int16).flatten().astype(np.float32) / 32768.0), input_sr=16000, batch_size=hf_batch_size).language else: @@ -362,8 +362,8 @@ def detect_language( logging.info(f"Error processing or transcribing Bazarr {audio_file.filename}: {e}") finally: - if f"Bazarr-detect-langauge-{random_name}" in files_to_transcribe: - files_to_transcribe.remove(f"Bazarr-detect-langauge-{random_name}") + if f"Bazarr-detect-language-{random_name}" in files_to_transcribe: + 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}