no more go

master
Bel LaPointe 2023-03-31 09:22:45 -06:00
parent 7e93939f3a
commit 48288689a6
4 changed files with 0 additions and 109 deletions

View File

@ -1,13 +0,0 @@
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
)

View File

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

View File

@ -1,12 +0,0 @@
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=

View File

@ -1,79 +0,0 @@
package main
import (
"bytes"
"fmt"
"io"
"log"
"net/http"
"os"
"strings"
"time"
"github.com/ggerganov/whisper.cpp/bindings/go/pkg/whisper"
"github.com/go-audio/wav"
)
func main() {
modelName := "small.en"
if v := os.Getenv("MODEL"); v != "" {
modelName = v
}
model, err := whisper.New("./models/ggml-" + modelName + ".bin")
if err != nil {
panic(err)
}
context, err := model.NewContext()
if err != nil {
panic(err)
}
context.SetThreads(4)
if err := http.ListenAndServe(":8080", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
b, _ := io.ReadAll(r.Body)
if result, err := transcribe(context, bytes.NewReader(b)); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
} else {
w.Write([]byte(result))
}
})); err != nil {
panic(err)
}
}
func transcribe(context whisper.Context, r io.ReadSeeker) (string, error) {
start := time.Now()
defer func() {
if os.Getenv("DEBUG") == "true" {
log.Printf("%0.1f to transcribe", time.Since(start).Seconds())
}
}()
var data []float32
dec := wav.NewDecoder(r)
if buf, err := dec.FullPCMBuffer(); err != nil {
return "", err
} else if dec.SampleRate != whisper.SampleRate {
return "", fmt.Errorf("sample rate %v != %v", dec.SampleRate, whisper.SampleRate)
} else if dec.NumChans != 1 {
return "", fmt.Errorf("chans %v != %v", dec.NumChans, 1)
} else {
data = buf.AsFloat32Buffer().Data
}
if err := context.Process(data, nil); err != nil {
return "", err
}
result := []string{}
for {
segment, err := context.NextSegment()
if err == io.EOF {
break
} else if err != nil {
return "", err
}
result = append(result, segment.Text)
}
return strings.Join(result, " "), nil
}