雾聪
2023-08-17 fbd9fbbde066a483fb903fe9c6c76fb95bc6fc2b
funasr/runtime/onnxruntime/include/offline-stream.h
@@ -8,14 +8,15 @@
#include "punc-model.h"
#include "vad-model.h"
namespace funasr {
class OfflineStream {
  public:
    OfflineStream(std::map<std::string, std::string>& model_path, int thread_num);
    ~OfflineStream(){};
    std::unique_ptr<VadModel> vad_handle;
    std::unique_ptr<Model> asr_handle;
    std::unique_ptr<PuncModel> punc_handle;
    std::unique_ptr<VadModel> vad_handle= nullptr;
    std::unique_ptr<Model> asr_handle= nullptr;
    std::unique_ptr<PuncModel> punc_handle= nullptr;
    bool UseVad(){return use_vad;};
    bool UsePunc(){return use_punc;}; 
    
@@ -25,4 +26,5 @@
};
OfflineStream *CreateOfflineStream(std::map<std::string, std::string>& model_path, int thread_num=1);
} // namespace funasr
#endif