dcos

游雁

2023-06-29 bc723ea200144bd6fa8a5dff4b9a780feda144fc

 funasr/runtime/onnxruntime/src/punc-model.cpp

@@ -1,10 +1,17 @@
#include "precomp.h"

PuncModel *CreatePuncModel(std::map<std::string, std::string>& model_path, int thread_num)
namespace funasr {
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;

@@ -17,3 +24,5 @@
    mm->InitPunc(punc_model_path, punc_config_path, thread_num);
    return mm;
}

} // namespace funasr