Merge branch 'main' into dev_smohan

yhliang

2023-05-11 1d1ef01b4e23630a99a3be7e9d1dce9550a793e9

 funasr/runtime/onnxruntime/include/offline-stream.h

New file
@@ -0,0 +1,30 @@
#ifndef OFFLINE_STREAM_H
#define OFFLINE_STREAM_H

#include <memory>
#include <string>
#include <map>
#include "model.h"
#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;
    bool UseVad(){return use_vad;};
    bool UsePunc(){return use_punc;}; 
    
  private:
    bool use_vad=false;
    bool use_punc=false;
};

OfflineStream *CreateOfflineStream(std::map<std::string, std::string>& model_path, int thread_num=1);
} // namespace funasr
#endif