vad-model-config.cc
1.4 KB
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
// sherpa-onnx/python/csrc/vad-model-config.h
//
// Copyright (c) 2023 Xiaomi Corporation
#include "sherpa-onnx/python/csrc/vad-model-config.h"
#include <string>
#include "sherpa-onnx/csrc/vad-model-config.h"
#include "sherpa-onnx/python/csrc/silero-vad-model-config.h"
#include "sherpa-onnx/python/csrc/ten-vad-model-config.h"
namespace sherpa_onnx {
void PybindVadModelConfig(py::module *m) {
PybindSileroVadModelConfig(m);
PybindTenVadModelConfig(m);
using PyClass = VadModelConfig;
py::class_<PyClass>(*m, "VadModelConfig")
.def(py::init<>())
.def(py::init<const SileroVadModelConfig &, const TenVadModelConfig &,
int32_t, int32_t, const std::string &, bool>(),
py::arg("silero_vad") = SileroVadModelConfig{},
py::arg("ten_vad") = TenVadModelConfig{},
py::arg("sample_rate") = 16000, py::arg("num_threads") = 1,
py::arg("provider") = "cpu", py::arg("debug") = false)
.def_readwrite("silero_vad", &PyClass::silero_vad)
.def_readwrite("ten_vad", &PyClass::ten_vad)
.def_readwrite("sample_rate", &PyClass::sample_rate)
.def_readwrite("num_threads", &PyClass::num_threads)
.def_readwrite("provider", &PyClass::provider)
.def_readwrite("debug", &PyClass::debug)
.def("__str__", &PyClass::ToString)
.def("validate", &PyClass::Validate);
}
} // namespace sherpa_onnx