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

Fix int casting on detect_language_length

Browse Source
This commit is contained in:
McCloudS
2024-04-23 08:42:42 -06:00
committed by GitHub
parent 0ea9a6429d
commit 1dd5c8c80b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
subgen.py
View File

@@ -466,7 +466,7 @@ def detect_language(
#encode: bool = Query(default=True, description="Encode audio first through ffmpeg") # This is always false from Bazarr #encode: bool = Query(default=True, description="Encode audio first through ffmpeg") # This is always false from Bazarr
): ):
detected_lang_code = "" # Initialize with an empty string detected_lang_code = "" # Initialize with an empty string
if 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:
start_model() start_model()
@@ -476,7 +476,7 @@ def detect_language(
task_queue.put(task_id) task_queue.put(task_id)
audio_data = np.frombuffer(audio_file.file.read(), np.int16).flatten().astype(np.float32) / 32768.0 audio_data = np.frombuffer(audio_file.file.read(), np.int16).flatten().astype(np.float32) / 32768.0
detected_lang_code = model.transcribe_stable(whisper.pad_or_trim(audio_data, detect_language_length * 16000), input_sr=16000).language detected_lang_code = model.transcribe_stable(whisper.pad_or_trim(audio_data, int(detect_language_length) * 16000), input_sr=16000).language
except Exception as e: except Exception as e:
logging.info(f"Error processing or transcribing Bazarr {audio_file.filename}: {e}") logging.info(f"Error processing or transcribing Bazarr {audio_file.filename}: {e}")