EngineX-Iluvatar/enginex_bi_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Add Go API examples for adding punctuations to text. (#1478)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-10-27 09:04:05 +08:00
committed by GitHub
parent 4a4659aa4f
commit 052b8645ba
9 changed files with 74 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/go/_internal/add-punctuation/go.mod Normal file
View File

@@ -0,0 +1,5 @@
module add-punctuation
go 1.12
replace github.com/k2-fsa/sherpa-onnx-go/sherpa_onnx => ../

1
scripts/go/_internal/add-punctuation/main.go Symbolic link
View File

@@ -0,0 +1 @@
../../../../go-api-examples/add-punctuation/main.go

1
scripts/go/_internal/add-punctuation/run.sh Symbolic link
View File

@@ -0,0 +1 @@
../../../../go-api-examples/add-punctuation/run.sh

12
scripts/go/sherpa_onnx.go
View File

@@ -1322,10 +1322,10 @@ func (sd *OfflineSpeakerDiarization) Process(samples []float32) []OfflineSpeaker
// For punctuation
// ============================================================
type OfflinePunctuationModelConfig struct {
Ct_transformer string
Num_threads C.int
Debug C.int // true to print debug information of the model
Provider string
CtTransformer string
NumThreads C.int
Debug C.int // true to print debug information of the model
Provider string
}
type OfflinePunctuationConfig struct {
@@ -1338,10 +1338,10 @@ type OfflinePunctuation struct {
func NewOfflinePunctuation(config *OfflinePunctuationConfig) *OfflinePunctuation {
cfg := C.struct_SherpaOnnxOfflinePunctuationConfig{}
cfg.model.ct_transformer = C.CString(config.Model.Ct_transformer)
cfg.model.ct_transformer = C.CString(config.Model.CtTransformer)
defer C.free(unsafe.Pointer(cfg.model.ct_transformer))
cfg.model.num_threads = config.Model.Num_threads
cfg.model.num_threads = config.Model.NumThreads
cfg.model.debug = config.Model.Debug
cfg.model.provider = C.CString(config.Model.Provider)
defer C.free(unsafe.Pointer(cfg.model.provider))