雾聪
2023-06-28 87c2c615e8bd2f4240b5f6f74f750e9e9dc52c17
add init for online punc
1个文件已修改
12 ■■■■ 已修改文件
funasr/runtime/onnxruntime/src/punc-model.cpp 12 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/runtime/onnxruntime/src/punc-model.cpp
@@ -1,11 +1,17 @@
#include "precomp.h"
namespace funasr {
PuncModel *CreatePuncModel(std::map<std::string, std::string>& model_path, int thread_num)
PuncModel *CreatePuncModel(std::map<std::string, std::string>& model_path, int thread_num, PUNC_TYPE type)
{
    PuncModel *mm;
    mm = new CTTransformer();
    if (type==PUNC_OFFLINE){
        mm = new CTTransformer();
    }else if(type==PUNC_ONLINE){
        mm = new CTTransformerOnline();
    }else{
        LOG(ERROR) << "Wrong PUNC TYPE";
        exit(-1);
    }
    string punc_model_path;
    string punc_config_path;