Просмотр исходного кода

update paraformer online websocket recipe

haoneng.lhn 2 лет назад
Родитель
Сommit
a1cc5a26a1
2 измененных файлов с 5 добавлено и 3 удалено
  1. 2 1
      egs_modelscope/asr/TEMPLATE/README.md
  2. 3 2
      funasr/runtime/python/websocket/wss_srv_asr.py

+ 2 - 1
egs_modelscope/asr/TEMPLATE/README.md

@@ -24,7 +24,8 @@ print(rec_result)
 inference_pipeline = pipeline(
 inference_pipeline = pipeline(
     task=Tasks.auto_speech_recognition,
     task=Tasks.auto_speech_recognition,
     model='damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-online',
     model='damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-online',
-    model_revision='v1.0.4'
+    model_revision='v1.0.6',
+    mode='paraformer_streaming'
     )
     )
 import soundfile
 import soundfile
 speech, sample_rate = soundfile.read("example/asr_example.wav")
 speech, sample_rate = soundfile.read("example/asr_example.wav")

+ 3 - 2
funasr/runtime/python/websocket/wss_srv_asr.py

@@ -58,7 +58,8 @@ inference_pipeline_asr_online = pipeline(
     model=args.asr_model_online,
     model=args.asr_model_online,
     ngpu=args.ngpu,
     ngpu=args.ngpu,
     ncpu=args.ncpu,
     ncpu=args.ncpu,
-    model_revision='v1.0.4')
+    model_revision='v1.0.6',
+    mode='paraformer_streaming')
 
 
 print("model loaded")
 print("model loaded")
 
 
@@ -207,4 +208,4 @@ if len(args.certfile)>0:
 else:
 else:
     start_server = websockets.serve(ws_serve, args.host, args.port, subprotocols=["binary"], ping_interval=None)
     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_until_complete(start_server)
-asyncio.get_event_loop().run_forever()
+asyncio.get_event_loop().run_forever()