alihan/whisper-asr-webservice
1
0
Fork 0
mirror of https://github.com/ahmetoner/whisper-asr-webservice.git synced 2023-04-14 03:48:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add prompt to ASR

Browse Source
This commit is contained in:
Vijaianand Mohan
2023-01-24 13:32:38 +05:30
parent d344302061
commit 31e1d85e64
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
app/webservice.py
View File

@@ -64,6 +64,7 @@ def transcribe(
audio_file: UploadFile = File(...),
task : Union[str, None] = Query(default="transcribe", enum=["transcribe", "translate"]),
language: Union[str, None] = Query(default=None, enum=LANGUAGE_CODES),
prompt: Union[str, None] = Query(default=None),
output : Union[str, None] = Query(default="json", enum=["json", "vtt", "srt"]),
):
@@ -108,11 +109,13 @@ def language_detection(
return result
def run_asr(file: BinaryIO, task: Union[str, None], language: Union[str, None] ):
def run_asr(file: BinaryIO, task: Union[str, None], language: Union[str, None], prompt: Union[str, None] ):
audio = load_audio(file)
options_dict = {"task" : task}
if language:
options_dict["language"] = language
if prompt:
options_dict["prompt"] = prompt
with model_lock:
result = model.transcribe(audio, **options_dict)