dasemu/Transcriptarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed 'langauge' typo

Browse Source
This commit is contained in:
McCloudS
2024-03-07 06:52:24 -07:00
committed by GitHub
parent 74e7942d70
commit d22aaa5db4
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
subgen/subgen.py
View File

@@ -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}