1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
| #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"
| #if !defined(__APPLE__)
| #include "itn-model.h"
| #endif
|
| namespace funasr {
| class OfflineStream {
| public:
| OfflineStream(std::map<std::string, std::string>& model_path, int thread_num, bool use_gpu=false, int batch_size=1);
| ~OfflineStream(){};
|
| std::unique_ptr<VadModel> vad_handle= nullptr;
| std::unique_ptr<Model> asr_handle= nullptr;
| std::unique_ptr<PuncModel> punc_handle= nullptr;
| #if !defined(__APPLE__)
| std::unique_ptr<ITNModel> itn_handle = nullptr;
| #endif
| bool UseVad(){return use_vad;};
| bool UsePunc(){return use_punc;};
| bool UseITN(){return use_itn;};
|
| private:
| bool use_vad=false;
| bool use_punc=false;
| bool use_itn=false;
| };
|
| OfflineStream *CreateOfflineStream(std::map<std::string, std::string>& model_path, int thread_num=1, bool use_gpu=false, int batch_size=1);
| } // namespace funasr
| #endif
|
|
