go bindings test

whisper-cpp-2023/go.mod

@@ -0,0 +1,13 @@
+module gogs.inhome.blapointe.com/stt.d/whisper-cpp-2023
+
+go 1.19
+
+require (
+	github.com/ggerganov/whisper.cpp/bindings/go v0.0.0-20230330172929-0a2d1210bcb9
+	github.com/go-audio/wav v1.1.0
+)
+
+require (
+	github.com/go-audio/audio v1.0.0 // indirect
+	github.com/go-audio/riff v1.0.0 // indirect
+)

whisper-cpp-2023/go.sh

@@ -0,0 +1,5 @@
+#! /bin/bash
+
+C_INCLUDE_PATH="$C_INCLUDE_PATH:$PWD" \
+LIBRARY_PATH="$LIBRARY_PATH:$PWD" \
+go "$@"

whisper-cpp-2023/go.sum

@@ -0,0 +1,12 @@
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/ggerganov/whisper.cpp/bindings/go v0.0.0-20230330172929-0a2d1210bcb9 h1:xzxmeGO16wpaBfXYxxCVpnOrdpUr7/E+VjHbrp5YVN8=
+github.com/ggerganov/whisper.cpp/bindings/go v0.0.0-20230330172929-0a2d1210bcb9/go.mod h1:QIjZ9OktHFG7p+/m3sMvrAJKKdWrr1fZIK0rM6HZlyo=
+github.com/go-audio/audio v1.0.0 h1:zS9vebldgbQqktK4H0lUqWrG8P0NxCJVqcj7ZpNnwd4=
+github.com/go-audio/audio v1.0.0/go.mod h1:6uAu0+H2lHkwdGsAY+j2wHPNPpPoeg5AaEFh9FlA+Zs=
+github.com/go-audio/riff v1.0.0 h1:d8iCGbDvox9BfLagY94fBynxSPHO80LmZCaOsmKxokA=
+github.com/go-audio/riff v1.0.0/go.mod h1:l3cQwc85y79NQFCRB7TiPoNiaijp6q8Z0Uv38rVG498=
+github.com/go-audio/wav v1.1.0 h1:jQgLtbqBzY7G+BM8fXF7AHUk1uHUviWS4X39d5rsL2g=
+github.com/go-audio/wav v1.1.0/go.mod h1:mpe9qfwbScEbkd8uybLuIpTgHyrISw/OTuvjUW2iGtE=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/stretchr/testify v1.8.1 h1:w7B6lhMri9wdJUVmEZPGGhZzrYTPvgJArz7wNPgYKsk=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=

whisper-cpp-2023/libwhisper.a

@@ -0,0 +1 @@
+git.d/libwhisper.a

whisper-cpp-2023/main.go

@@ -0,0 +1,60 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"os"
+
+	"github.com/ggerganov/whisper.cpp/bindings/go/pkg/whisper"
+	"github.com/go-audio/wav"
+)
+
+func main() {
+	p := os.Args[1]
+	f, err := os.Open(p)
+	if err != nil {
+		panic(err)
+	}
+	defer f.Close()
+
+	var data []float32
+	dec := wav.NewDecoder(f)
+	if buf, err := dec.FullPCMBuffer(); err != nil {
+		panic(err)
+	} else if dec.SampleRate != whisper.SampleRate {
+		panic(dec.SampleRate)
+	} else if dec.NumChans != 1 {
+		panic(dec.NumChans)
+	} else {
+		data = buf.AsFloat32Buffer().Data
+	}
+
+	model, err := whisper.New("./models/ggml-small.en.bin")
+	if err != nil {
+		panic(err)
+	}
+	context, err := model.NewContext()
+	if err != nil {
+		panic(err)
+	}
+	context.SetThreads(4)
+
+	context.ResetTimings()
+	if err := context.Process(data, func(segment whisper.Segment) {
+		//log.Printf("%+v", segment)
+	}); err != nil {
+		panic(err)
+	}
+	context.PrintTimings()
+
+	for {
+		segment, err := context.NextSegment()
+		if err == io.EOF {
+			break
+		} else if err != nil {
+			panic(err)
+		}
+		fmt.Printf("%s ", segment.Text)
+	}
+	fmt.Printf("\n")
+}

whisper-cpp-2023/whisper.h

@@ -0,0 +1 @@
+git.d/whisper.h