FunASR/runtime/onnxruntime/include/tpass-online-stream.h
2024-08-01 14:52:07 +08:00

25 lines
673 B
C++

#ifndef TPASS_ONLINE_STREAM_H
#define TPASS_ONLINE_STREAM_H
#include <memory>
#include "tpass-stream.h"
#include "model.h"
#include "vad-model.h"
namespace funasr {
class TpassOnlineStream {
public:
TpassOnlineStream(TpassStream* tpass_stream, std::vector<int> chunk_size);
~TpassOnlineStream(){};
std::unique_ptr<VadModel> vad_online_handle = nullptr;
std::unique_ptr<Model> asr_online_handle = nullptr;
//for sv-cam
std::vector<std::vector<float>> voice_feats;
};
TpassOnlineStream* CreateTpassOnlineStream(void* tpass_stream, std::vector<int> chunk_size);
void TpassOnlineCacheReset(void* tpass_online_stream);
} // namespace funasr
#endif