zhaomingwork
2023-05-24 b8dd1e310b631cba7f8124ff30783ea60026b724
funasr/runtime/python/websocket/ws_server_offline.py
@@ -5,6 +5,7 @@
import logging
import tracemalloc
import numpy as np
import ssl
from parse_args import args
from modelscope.pipelines import pipeline
@@ -147,9 +148,16 @@
                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
start_server = websockets.serve(ws_serve, args.host, args.port, subprotocols=["binary"], ping_interval=None)
   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()