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

support translate endpoint

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

11
whisper_fastapi.py
View File

@@ -298,10 +298,11 @@ async def translateapi(
@app.post("/v1/audio/transcriptions") @app.post("/v1/audio/transcriptions")
@app.post("/v1/audio/translations") @app.post("/v1/audio/translations")
async def transcription( async def transcription(
request: Request,
file: UploadFile = File(...), file: UploadFile = File(...),
prompt: str = Form(""), prompt: str = Form(""),
response_format: str = Form("json"), response_format: str = Form("json"),
task: str = Form("transcribe"), task: str = Form(""),
language: str = Form("und"), language: str = Form("und"),
vad_filter: bool = Form(False), vad_filter: bool = Form(False),
repetition_penalty: float = Form(1.0), repetition_penalty: float = Form(1.0),
@@ -311,6 +312,14 @@ async def transcription(
User upload audio file in multipart/form-data format and receive transcription in response User upload audio file in multipart/form-data format and receive transcription in response
""" """
if not task:
if request.url.path == '/v1/audio/transcriptions':
task = "transcribe"
elif request.url.path == '/v1/audio/translations':
task = "translate"
else:
raise HTTPException(400, "task parameter is required")
# timestamp as filename, keep original extension # timestamp as filename, keep original extension
generator, info = stream_builder( generator, info = stream_builder(
audio=io.BytesIO(file.file.read()), audio=io.BytesIO(file.file.read()),