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

Fixed urls ending in /

Browse Source
This commit is contained in:
McCloudS
2024-03-10 09:31:03 -06:00
committed by GitHub
parent de843aacf1
commit d5e3acc5fd
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
subgen/subgen.py
View File

@@ -298,6 +298,7 @@ def batch(
transcribe_existing(directory, forceLanguage) transcribe_existing(directory, forceLanguage)
# idea and some code for asr and detect language from https://github.com/ahmetoner/whisper-asr-webservice # idea and some code for asr and detect language from https://github.com/ahmetoner/whisper-asr-webservice
@app.post("//asr")
@app.post("/asr") @app.post("/asr")
def asr( def asr(
task: Union[str, None] = Query(default="transcribe", enum=["transcribe", "translate"]), task: Union[str, None] = Query(default="transcribe", enum=["transcribe", "translate"]),
@@ -316,7 +317,7 @@ def asr(
start_time = time.time() start_time = time.time()
start_model() start_model()
files_to_transcribe.insert(0, f"Bazarr-detect-language-{random_name}") files_to_transcribe.insert(0, f"Bazarr-asr-{random_name}")
if(hf_transformers): if(hf_transformers):
result = model.transcribe(np.frombuffer(audio_file.file.read(), np.int16).flatten().astype(np.float32) / 32768.0, task=task, input_sr=16000, language=language, batch_size=hf_batch_size, progress_callback=progress) result = model.transcribe(np.frombuffer(audio_file.file.read(), np.int16).flatten().astype(np.float32) / 32768.0, task=task, input_sr=16000, language=language, batch_size=hf_batch_size, progress_callback=progress)
else: else:
@@ -328,8 +329,8 @@ def asr(
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}")
finally: finally:
if f"Bazarr-detect-language-{random_name}" in files_to_transcribe: if f"Bazarr-asr-{random_name}" in files_to_transcribe:
files_to_transcribe.remove(f"Bazarr-detect-language-{random_name}") files_to_transcribe.remove(f"Bazarr-asr-{random_name}")
delete_model() delete_model()
if result: if result:
return StreamingResponse( return StreamingResponse(
@@ -341,6 +342,7 @@ def asr(
else: else:
return return
@app.post("//detect-language")
@app.post("/detect-language") @app.post("/detect-language")
def detect_language( def detect_language(
audio_file: UploadFile = File(...), audio_file: UploadFile = File(...),