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

Update subgen.py

Browse Source
This commit is contained in:
McCloudS
2024-02-09 10:14:57 -07:00
committed by GitHub
parent 1606923e18
commit e2dd80e3dd
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
subgen/subgen.py
View File

@@ -272,7 +272,7 @@ def gen_subtitles(file_path: str, transcribe_or_translate_str: str, front=True)
logging.debug(f"{file_path} already has an internal sub we want, skipping generation") logging.debug(f"{file_path} already has an internal sub we want, skipping generation")
return f"{file_path} already has an internal sub we want, skipping generation" return f"{file_path} already has an internal sub we want, skipping generation"
elif os.path.exists(get_file_name_without_extension(file_path) + subextension): elif os.path.exists(get_file_name_without_extension(file_path) + subextension):
print(f"{file_path} already has a subgen created for this, skipping it") logging.debug(f"{file_path} already has a subgen created for this, skipping it")
return f"{file_path} already has a subgen created for this, skipping it" return f"{file_path} already has a subgen created for this, skipping it"
if front: if front:
@@ -288,6 +288,7 @@ def gen_subtitles(file_path: str, transcribe_or_translate_str: str, front=True)
start_model() start_model()
if(force_detected_language_to): if(force_detected_language_to):
logging.debug(f"Forcing detected audio language to {force_detected_language_to}")
result = model.transcribe_stable(file_path, language=force_detected_language_to, task=transcribe_or_translate_str) result = model.transcribe_stable(file_path, language=force_detected_language_to, task=transcribe_or_translate_str)
else: else:
result = model.transcribe_stable(file_path, task=transcribe_or_translate_str) result = model.transcribe_stable(file_path, task=transcribe_or_translate_str)