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-03-14 09:27:44 -06:00
committed by GitHub
parent 5b6b04de23
commit 67ef9a03da
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
subgen/subgen.py
View File

@@ -312,10 +312,10 @@ def asr(
output: Union[str, None] = Query(default="srt", enum=["txt", "vtt", "srt", "tsv", "json"]), 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 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: try:
logging.info(f"Transcribing file from Bazarr/ASR webhook") logging.info(f"Transcribing file from Bazarr/ASR webhook")
result = None result = None
random_name = ''.join(random.choices(string.ascii_letters + string.digits, k=6))
start_time = time.time() start_time = time.time()
start_model() start_model()
@@ -352,8 +352,8 @@ 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
random_name = ''.join(random.choices(string.ascii_letters + string.digits, k=6))
try: try:
random_name = ''.join(random.choices(string.ascii_letters + string.digits, k=6))
start_model() start_model()
files_to_transcribe.insert(0, f"Bazarr-detect-language-{random_name}") files_to_transcribe.insert(0, f"Bazarr-detect-language-{random_name}")