//go:build linux
|
|
package engine
|
|
import (
|
"fmt"
|
"voicesnap/internal/logger"
|
"voicesnap/internal/model"
|
|
_ "github.com/k2-fsa/sherpa-onnx-go-linux"
|
sherpa "github.com/k2-fsa/sherpa-onnx-go/sherpa_onnx"
|
)
|
|
type sherpaEngine struct {
|
recognizer *sherpa.OfflineRecognizer
|
hwInfo string
|
}
|
|
func newPlatformEngine(resolved model.ResolvedModel) (Engine, error) {
|
// Linux: CPU only
|
config := sherpa.OfflineRecognizerConfig{}
|
config.FeatConfig.SampleRate = 16000
|
config.FeatConfig.FeatureDim = 80
|
config.ModelConfig.SenseVoice.Model = resolved.Files["model"]
|
config.ModelConfig.SenseVoice.UseInverseTextNormalization = 1
|
config.ModelConfig.Tokens = resolved.Files["tokens"]
|
config.ModelConfig.NumThreads = resolved.Profile.NumThreads
|
config.ModelConfig.Provider = "cpu"
|
config.DecodingMethod = "greedy_search"
|
|
recognizer := sherpa.NewOfflineRecognizer(&config)
|
if recognizer != nil {
|
info := "SenseVoice ยท CPU"
|
logger.Info("Engine initialized: %s", info)
|
return &sherpaEngine{
|
recognizer: recognizer,
|
hwInfo: info,
|
}, nil
|
}
|
|
return nil, fmt.Errorf("failed to initialize sherpa-onnx on Linux")
|
}
|
|
func (e *sherpaEngine) Recognize(samples []float32) (string, error) {
|
stream := sherpa.NewOfflineStream(e.recognizer)
|
defer sherpa.DeleteOfflineStream(stream)
|
|
stream.AcceptWaveform(16000, samples)
|
|
e.recognizer.Decode(stream)
|
result := stream.GetResult()
|
|
return result.Text, nil
|
}
|
|
func (e *sherpaEngine) HardwareInfo() string {
|
return e.hwInfo
|
}
|
|
func (e *sherpaEngine) Close() {
|
if e.recognizer != nil {
|
sherpa.DeleteOfflineRecognizer(e.recognizer)
|
e.recognizer = nil
|
}
|
}
|
