heimoshuiyu/whisper-fastapi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: initial_prompt params

Browse Source
This commit is contained in:
黑墨水鱼
2024-09-13 16:13:19 +08:00
parent 71bde08b17
commit fab1ec9d03
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
whisper_fastapi.py
View File

@@ -127,7 +127,7 @@ def stream_builder(
language=language, language=language,
task=task, task=task,
vad_filter=vad_filter, vad_filter=vad_filter,
initial_prompt=initial_prompt, initial_prompt=initial_prompt if initial_prompt else None,
word_timestamps=True, word_timestamps=True,
repetition_penalty=repetition_penalty, repetition_penalty=repetition_penalty,
) )
@@ -296,6 +296,7 @@ async def translateapi(
@app.post("/v1/audio/transcriptions") @app.post("/v1/audio/transcriptions")
@app.post("/v1/audio/translations")
async def transcription( async def transcription(
file: UploadFile = File(...), file: UploadFile = File(...),
prompt: str = Form(""), prompt: str = Form(""),
@@ -315,6 +316,7 @@ async def transcription(
audio=io.BytesIO(file.file.read()), audio=io.BytesIO(file.file.read()),
task=task, task=task,
vad_filter=vad_filter, vad_filter=vad_filter,
initial_prompt=prompt,
language=None if language == "und" else language, language=None if language == "und" else language,
repetition_penalty=repetition_penalty, repetition_penalty=repetition_penalty,
) )