zhaomingwork
2023-05-24 b8dd1e310b631cba7f8124ff30783ea60026b724
funasr/runtime/python/websocket/ws_server_2pass.py
@@ -5,7 +5,7 @@
import logging
import tracemalloc
import numpy as np
import ssl
from parse_args import args
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
@@ -191,7 +191,16 @@
                message = json.dumps({"mode": "2pass-online", "text": rec_result["text"], "wav_name": websocket.wav_name})
                await websocket.send(message)
if len(args.certfile)>0:
   ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
   # Generate with Lets Encrypt, copied to this location, chown to current user and 400 permissions
   ssl_cert = args.certfile
   ssl_key = args.keyfile
   ssl_context.load_cert_chain(ssl_cert, keyfile=ssl_key)
   start_server = websockets.serve(ws_serve, args.host, args.port, subprotocols=["binary"], ping_interval=None,ssl=ssl_context)
else:
start_server = websockets.serve(ws_serve, args.host, args.port, subprotocols=["binary"], ping_interval=None)
asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()