no more go
parent
7e93939f3a
commit
48288689a6
|
|
@ -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
|
|
||||||
)
|
|
||||||
|
|
@ -1,5 +0,0 @@
|
||||||
#! /bin/bash
|
|
||||||
|
|
||||||
C_INCLUDE_PATH="$C_INCLUDE_PATH:$PWD" \
|
|
||||||
LIBRARY_PATH="$LIBRARY_PATH:$PWD" \
|
|
||||||
go "$@"
|
|
||||||
|
|
@ -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=
|
|
||||||
|
|
@ -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
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue