From 9e1de94db4a79a00e88df9e6bf3b1a375f1cbca4 Mon Sep 17 00:00:00 2001 From: McCloudS <64094529+McCloudS@users.noreply.github.com> Date: Thu, 14 Mar 2024 09:32:23 -0600 Subject: [PATCH] Update subgen.py --- subgen/subgen.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/subgen/subgen.py b/subgen/subgen.py index c21c35a..2e1f18f 100644 --- a/subgen/subgen.py +++ b/subgen/subgen.py @@ -312,10 +312,10 @@ def asr( output: Union[str, None] = Query(default="srt", enum=["txt", "vtt", "srt", "tsv", "json"]), word_timestamps: bool = Query(default=False, description="Word level timestamps") #not used by Bazarr ): - random_name = ''.join(random.choices(string.ascii_letters + string.digits, k=6)) try: logging.info(f"Transcribing file from Bazarr/ASR webhook") result = None + random_name = random.choices("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890", k=6) start_time = time.time() start_model() @@ -352,10 +352,9 @@ def detect_language( #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 - random_name = ''.join(random.choices(string.ascii_letters + string.digits, k=6)) try: start_model() - + random_name = random.choices("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890", k=6) files_to_transcribe.insert(0, f"Bazarr-detect-language-{random_name}") audio_data = np.frombuffer(audio_file.file.read(), np.int16).flatten().astype(np.float32) / 32768.0 if(hf_transformers):