mirror of
https://github.com/alphacep/vosk-api.git
synced 2026-01-09 07:32:34 +08:00
55 lines
790 B
Go
55 lines
790 B
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
|
|
vosk "github.com/alphacep/vosk-api/go"
|
|
)
|
|
|
|
func main() {
|
|
var filename string
|
|
flag.StringVar(&filename, "f", "", "file to transcribe")
|
|
flag.Parse()
|
|
|
|
vosk.GPUInit()
|
|
|
|
model, err := vosk.NewBatchModel("model")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
rec, err := vosk.NewBatchRecognizer(model, 16000.0)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
file, err := os.Open(filename)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer file.Close()
|
|
|
|
buf := make([]byte, 8000)
|
|
|
|
for {
|
|
if _, err := file.Read(buf); err != nil {
|
|
if err != io.EOF {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
break
|
|
}
|
|
rec.AcceptWaveform(buf)
|
|
model.Wait()
|
|
if rec.FrontResult() != "" {
|
|
fmt.Println(rec.FrontResult())
|
|
rec.Pop()
|
|
}
|
|
}
|
|
// Is this needed? rec.FinishStream()
|
|
}
|