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

Remove print statements and unnecessary code

Browse Source
This commit is contained in:
黑墨水鱼
2024-08-04 16:59:16 +08:00
parent 1c93201250
commit f71ef945db
1 changed files with 0 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
whisper_fastapi.py
View File

@@ -182,15 +182,11 @@ async def konele_ws(
# convert lang code format (eg. en-US to en) # convert lang code format (eg. en-US to en)
lang = lang.split("-")[0] lang = lang.split("-")[0]
print("WebSocket client connected, lang is", lang)
print("content-type is", content_type)
data = b"" data = b""
while True: while True:
try: try:
data += await websocket.receive_bytes() data += await websocket.receive_bytes()
print("Received data:", len(data), data[-10:])
if data[-3:] == b"EOS": if data[-3:] == b"EOS":
print("End of speech")
break break
except: except:
break break
@@ -223,7 +219,6 @@ async def konele_ws(
result = build_json_result(generator, info) result = build_json_result(generator, info)
text = result.get("text", "") text = result.get("text", "")
print("result", text)
await websocket.send_json( await websocket.send_json(
{ {
@@ -246,14 +241,12 @@ async def translateapi(
vad_filter: bool = False, vad_filter: bool = False,
): ):
content_type = request.headers.get("Content-Type", "") content_type = request.headers.get("Content-Type", "")
print("downloading request file", content_type)
# convert lang code format (eg. en-US to en) # convert lang code format (eg. en-US to en)
lang = lang.split("-")[0] lang = lang.split("-")[0]
splited = [i.strip() for i in content_type.split(",") if "=" in i] splited = [i.strip() for i in content_type.split(",") if "=" in i]
info = {k: v for k, v in (i.split("=") for i in splited)} info = {k: v for k, v in (i.split("=") for i in splited)}
print(info)
channels = int(info.get("channels", "1")) channels = int(info.get("channels", "1"))
rate = int(info.get("rate", "16000")) rate = int(info.get("rate", "16000"))
@@ -287,7 +280,6 @@ async def translateapi(
result = build_json_result(generator, info) result = build_json_result(generator, info)
text = result.get("text", "") text = result.get("text", "")
print("result", text)
return { return {
"status": 0, "status": 0,