游雁
2023-10-10 f974935484d5d8eb37b36eb2646816c02a41184c
funasr/runtime/onnxruntime/src/paraformer.h
@@ -33,7 +33,6 @@
        vector<const char*> hw_m_szInputNames;
        vector<const char*> hw_m_szOutputNames;
        bool use_hotword;
        std::string language="zh-cn";
    public:
        Paraformer();
@@ -55,6 +54,7 @@
        string PostProcess(std::vector<string> &raw_char, std::vector<std::vector<float>> &timestamp_list);
        string Rescoring();
        string GetLang(){return language;};
        knf::FbankOptions fbank_opts_;
        vector<float> means_list_;
@@ -71,6 +71,8 @@
        vector<const char*> m_szInputNames;
        vector<const char*> m_szOutputNames;
        std::string language="zh-cn";
        // paraformer-online
        std::shared_ptr<Ort::Session> encoder_session_ = nullptr;
        std::shared_ptr<Ort::Session> decoder_session_ = nullptr;