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

add: initial_prompt args

Browse Source
This commit is contained in:
黑墨水鱼
2023-11-15 16:26:41 +08:00
parent 86691e9883
commit c3838dcb3f
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
whisper_fastapi.py
View File

@@ -125,6 +125,7 @@ async def konele_ws(
websocket: WebSocket, websocket: WebSocket,
task: Literal["transcribe", "translate"] = "transcribe", task: Literal["transcribe", "translate"] = "transcribe",
lang: str = "und", lang: str = "und",
initial_prompt: str = "",
): ):
await websocket.accept() await websocket.accept()
print("WebSocket client connected, lang is", lang) print("WebSocket client connected, lang is", lang)
@@ -151,7 +152,7 @@ async def konele_ws(
buffer.writeframes(data) buffer.writeframes(data)
file_obj.seek(0) file_obj.seek(0)
options = get_options() options = get_options(initial_prompt=initial_prompt)
result = transcriber.inference( result = transcriber.inference(
audio=file_obj, audio=file_obj,
@@ -182,6 +183,7 @@ async def translateapi(
request: Request, request: Request,
task: Literal["transcribe", "translate"] = "transcribe", task: Literal["transcribe", "translate"] = "transcribe",
lang: str = "und", lang: str = "und",
initial_prompt: str = "",
): ):
content_type = request.headers.get("Content-Type", "") content_type = request.headers.get("Content-Type", "")
print("downloading request file", content_type) print("downloading request file", content_type)
@@ -205,7 +207,7 @@ async def translateapi(
buffer.writeframes(body) buffer.writeframes(body)
file_obj.seek(0) file_obj.seek(0)
options = get_options() options = get_options(initial_prompt=initial_prompt)
result = transcriber.inference( result = transcriber.inference(
audio=file_obj, audio=file_obj,