Go bindings for Whisper
This package provides Go bindings for whisper.cpp. They have been tested on:
- Darwin (OS X) 12.6 on x64_64
- Debian Linux on arm64
- Fedora Linux on x86_64
The "low level" bindings are in the bindings/go directory and there is a more
Go-style package in the bindings/go/pkg/whisper directory. The most simple usage
is as follows:
import (
"github.com/ggerganov/whisper.cpp/bindings/go/pkg/whisper"
)
func main() {
var modelpath string
var samples []float32
model, err := whisper.New(modelpath)
if err != nil {
panic(err)
}
defer model.Close()
context, err := model.NewContext()
if err != nil {
panic(err)
}
if err := context.Process(samples, nil, nil, nil); err != nil {
return err
}
for {
segment, err := context.NextSegment()
if err != nil {
break
}
fmt.Printf("[%6s->%6s] %s\n", segment.Start, segment.End, segment.Text)
}
}
Building & Testing
In order to build, you need to have the Go compiler installed. You can get it from here. Run the tests with:
git clone https://github.com/ggml-org/whisper.cpp.git
cd whisper.cpp/bindings/go
make test
This will compile a static libwhisper.a in a build folder, download a model file, then run the tests. To build the examples:
make examples
To build using cuda support add GGML_CUDA=1:
GGML_CUDA=1 make examples
The examples are placed in the build directory. Once built, you can download all the models with the following command:
./build/go-model-download -out models
And you can then test a model against samples with the following command:
./build/go-whisper -model models/ggml-tiny.en.bin samples/jfk.wav
Using the bindings
To use the bindings in your own software,
- Import
github.com/ggerganov/whisper.cpp/bindings/go/pkg/whisper (or github.com/ggerganov/whisper.cpp/bindings/go into your package;
- Compile
libwhisper.a (you can use make whisper in the bindings/go directory);
- Link your go binary against whisper by setting the environment variables
C_INCLUDE_PATH and LIBRARY_PATH
to point to the whisper.h file directory and libwhisper.a file directory respectively.
Look at the Makefile in the bindings/go directory for an example.
The API Documentation:
Getting help:
- Follow the discussion for the go bindings here
License
The license for the Go bindings is the same as the license for the rest of the whisper.cpp project, which is the MIT License. See the LICENSE file for more details.