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

Clarified when language is being forced

Browse Source
This commit is contained in:
McCloudS
2024-08-05 21:29:28 -06:00
committed by GitHub
parent b727394155
commit b8e3ea026d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
subgen.py
View File

@@ -439,6 +439,7 @@ async def asr(
if force_detected_language_to: if force_detected_language_to:
language = force_detected_language_to language = force_detected_language_to
logging.info(f"ENV FORCE_DETECTED_LANGUAGE_TO is set: Forcing detected language to {force_language}")
start_time = time.time() start_time = time.time()
start_model() start_model()
@@ -482,6 +483,7 @@ async def detect_language(
): ):
detected_language = "" # Initialize with an empty string detected_language = "" # Initialize with an empty string
language_code = "" # Initialize with an empty string language_code = "" # Initialize with an empty string
logging.info(f"ENV FORCE_DETECTED_LANGUAGE_TO is set to {force_language}, the language detected from this call will be ignored in /ASR calls")
if int(detect_language_length) != 30: if int(detect_language_length) != 30:
logging.info(f"Detect language is set to detect on the first {detect_language_length} seconds of the audio.") logging.info(f"Detect language is set to detect on the first {detect_language_length} seconds of the audio.")
try: try:
@@ -547,10 +549,10 @@ def gen_subtitles(file_path: str, transcription_type: str, force_language=None)
start_model() start_model()
if force_language: if force_language:
logging.info(f"Forcing language to {force_language}") logging.info(f"Forcing detected language to {force_language} from /batch endpoint")
elif force_detected_language_to: elif force_detected_language_to:
force_language = force_detected_language_to force_language = force_detected_language_to
logging.info(f"Forcing language to {force_language}") logging.info(f"ENV FORCE_DETECTED_LANGUAGE_TO is set: Forcing detected language to {force_language}")
if custom_regroup: if custom_regroup:
result = model.transcribe_stable(file_path, language=force_language, task=transcription_type, result = model.transcribe_stable(file_path, language=force_language, task=transcription_type,