Merge pull request #830 from alibaba-damo-academy/dev_ts

Xian Shi

2023-08-10 e2b3edec45fe1bfd76493acc366df971e89e7ae2

 funasr/runtime/websocket/funasr-wss-server.cpp

@@ -79,7 +79,7 @@
    TCLAP::ValueArg<int> decoder_thread_num(
        "", "decoder-thread-num", "decoder thread num", false, 8, "int");
    TCLAP::ValueArg<int> model_thread_num("", "model-thread-num",
                                          "model thread num", false, 1, "int");
                                          "model thread num", false, 4, "int");

    TCLAP::ValueArg<std::string> certfile("", "certfile", 
        "default: ../../../ssl_key/server.crt, path of certficate for WSS connection. if it is empty, it will be in WS mode.", 
@@ -125,12 +125,6 @@
    try{
        std::string s_download_model_dir = download_model_dir.getValue();

        // 判断 s_download_model_dir 是否存在，目前用不到
        if (access(s_download_model_dir.c_str(), F_OK) != 0){
            LOG(ERROR) << s_download_model_dir << " do not exists."; 
            exit(-1);
        }

        std::string s_vad_path = model_path[VAD_DIR];
        std::string s_vad_quant = model_path[VAD_QUANT];
        std::string s_asr_path = model_path[MODEL_DIR];
@@ -158,7 +152,7 @@
                
            int ret = system(python_cmd_vad.c_str());
            if(ret !=0){
                LOG(INFO) << "Failed to download model from modelscope. If you set local model path, you can ignore the errors.";
                LOG(INFO) << "Failed to download model from modelscope. If you set local vad model path, you can ignore the errors.";
            }
            down_vad_model = down_vad_path+"/model_quant.onnx";
            if(s_vad_quant=="false" || s_vad_quant=="False" || s_vad_quant=="FALSE"){
@@ -194,7 +188,7 @@
                
            int ret = system(python_cmd_asr.c_str());
            if(ret !=0){
                LOG(INFO) << "Failed to download model from modelscope. If you set local model path, you can ignore the errors.";
                LOG(INFO) << "Failed to download model from modelscope. If you set local asr model path, you can ignore the errors.";
            }
            down_asr_model = down_asr_path+"/model_quant.onnx";
            if(s_asr_quant=="false" || s_asr_quant=="False" || s_asr_quant=="FALSE"){
@@ -230,7 +224,7 @@
                
            int ret = system(python_cmd_punc.c_str());
            if(ret !=0){
                LOG(INFO) << "Failed to download model from modelscope. If you set local model path, you can ignore the errors.";
                LOG(INFO) << "Failed to download model from modelscope. If you set local punc model path, you can ignore the errors.";
            }
            down_punc_model = down_punc_path+"/model_quant.onnx";
            if(s_punc_quant=="false" || s_punc_quant=="False" || s_punc_quant=="FALSE"){