Răsfoiți Sursa

update funasr-wss-server-2pass args

雾聪 2 ani în urmă
părinte
comite
0d9b907b37

+ 3 - 0
funasr/runtime/onnxruntime/src/funasrruntime.cpp

@@ -379,6 +379,9 @@ extern "C" {
 		while(audio->FetchTpass(frame) > 0){
 			string msg = asr_handle->Forward(frame->data, frame->len, frame->is_final);
 			string msg_punc = punc_online_handle->AddPunc(msg.c_str(), punc_cache[1]);
+			if(input_finished){
+				msg_punc += "。";
+			}
 			p_result->tpass_msg = msg_punc;
 			if(frame != NULL){
 				delete frame;

+ 6 - 6
funasr/runtime/websocket/funasr-wss-server-2pass.cpp

@@ -32,9 +32,9 @@ int main(int argc, char* argv[]) {
         "/workspace/models", "string");
     TCLAP::ValueArg<std::string> offline_model_dir(
         "", OFFLINE_MODEL_DIR,
-        "default: /workspace/models/offline_asr, the asr model path, which "
+        "default: damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-onnx, the asr model path, which "
         "contains model_quant.onnx, config.yaml, am.mvn",
-        false, "/workspace/models/offline_asr", "string");
+        false, "damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-onnx", "string");
     TCLAP::ValueArg<std::string> online_model_dir(
         "", ONLINE_MODEL_DIR,
         "default: damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-online-onnx, the asr model path, which "
@@ -57,9 +57,9 @@ int main(int argc, char* argv[]) {
         false, "true", "string");
     TCLAP::ValueArg<std::string> vad_dir(
         "", VAD_DIR,
-        "default: /workspace/models/vad, the vad model path, which contains "
+        "default: damo/speech_fsmn_vad_zh-cn-16k-common-onnx, the vad model path, which contains "
         "model_quant.onnx, vad.yaml, vad.mvn",
-        false, "/workspace/models/vad", "string");
+        false, "damo/speech_fsmn_vad_zh-cn-16k-common-onnx", "string");
     TCLAP::ValueArg<std::string> vad_revision(
         "", "vad-revision", "VAD model revision", false, "v1.2.0", "string");
     TCLAP::ValueArg<std::string> vad_quant(
@@ -69,9 +69,9 @@ int main(int argc, char* argv[]) {
         false, "true", "string");
     TCLAP::ValueArg<std::string> punc_dir(
         "", PUNC_DIR,
-        "default: /workspace/models/punc, the punc model path, which contains "
+        "default: damo/punc_ct-transformer_zh-cn-common-vad_realtime-vocab272727-onnx, the punc model path, which contains "
         "model_quant.onnx, punc.yaml",
-        false, "/workspace/models/punc", "string");
+        false, "damo/punc_ct-transformer_zh-cn-common-vad_realtime-vocab272727-onnx", "string");
     TCLAP::ValueArg<std::string> punc_revision(
         "", "punc-revision", "PUNC model revision", false, "v1.0.2", "string");
     TCLAP::ValueArg<std::string> punc_quant(