lyblsgo
2023-04-23 55708e7cebaedefc5f69d61f157993da41848b8f
funasr/runtime/onnxruntime/src/paraformer_onnx.h
@@ -14,6 +14,7 @@
        knf::FbankOptions fbank_opts;
        std::unique_ptr<FsmnVad> vadHandle;
        std::unique_ptr<CTTransformer> puncHandle;
        Vocab* vocab;
        vector<float> means_list;
@@ -32,7 +33,6 @@
        Ort::Env env_;
        Ort::SessionOptions sessionOptions;
        vector<string> m_strInputNames, m_strOutputNames;
        vector<const char*> m_szInputNames;
        vector<const char*> m_szOutputNames;
@@ -45,6 +45,7 @@
        string forward(float* din, int len, int flag);
        string rescoring();
        std::vector<std::vector<int>> vad_seg(std::vector<float>& pcm_data);
        string AddPunc(const char* szInput);
    };