funasr/runtime/onnxruntime/src/ct-transformer.h
@@ -5,6 +5,7 @@ #pragma once namespace funasr { class CTTransformer : public PuncModel { /** * Author: Speech Lab of DAMO Academy, Alibaba Group @@ -28,5 +29,6 @@ void InitPunc(const std::string &punc_model, const std::string &punc_config, int thread_num); ~CTTransformer(); vector<int> Infer(vector<int32_t> input_data); string AddPunc(const char* sz_input); string AddPunc(const char* sz_input, std::string language="zh-cn"); }; } // namespace funasr