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