#pragma once 

class CTTransformer {
/**
 * Author: Speech Lab of DAMO Academy, Alibaba Group
 * CT-Transformer: Controllable time-delay transformer for real-time punctuation prediction and disfluency detection
 * https://arxiv.org/pdf/2003.01309.pdf
*/

private:

	CTokenizer m_tokenizer;
	vector<string> m_strInputNames, m_strOutputNames;
	vector<const char*> m_szInputNames;
	vector<const char*> m_szOutputNames;

	std::shared_ptr<Ort::Session> m_session;
    Ort::Env env_;
    Ort::SessionOptions session_options;
public:

	CTTransformer(const char* sz_model_dir, int thread_num);
	~CTTransformer();
	vector<int>  Infer(vector<int64_t> input_data);
	string AddPunc(const char* sz_input);
};