From fab1ec9d03cf8ff29ae2beb64b352ac27f85c3de Mon Sep 17 00:00:00 2001 From: heimoshuiyu Date: Fri, 13 Sep 2024 16:13:19 +0800 Subject: [PATCH] fix: initial_prompt params --- whisper_fastapi.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/whisper_fastapi.py b/whisper_fastapi.py index 579e197..60eac76 100644 --- a/whisper_fastapi.py +++ b/whisper_fastapi.py @@ -127,7 +127,7 @@ def stream_builder( language=language, task=task, vad_filter=vad_filter, - initial_prompt=initial_prompt, + initial_prompt=initial_prompt if initial_prompt else None, word_timestamps=True, repetition_penalty=repetition_penalty, ) @@ -296,6 +296,7 @@ async def translateapi( @app.post("/v1/audio/transcriptions") +@app.post("/v1/audio/translations") async def transcription( file: UploadFile = File(...), prompt: str = Form(""), @@ -315,6 +316,7 @@ async def transcription( audio=io.BytesIO(file.file.read()), task=task, vad_filter=vad_filter, + initial_prompt=prompt, language=None if language == "und" else language, repetition_penalty=repetition_penalty, )