30 lines
524 B
Go
30 lines
524 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/labstack/echo/v4"
|
||
|
|
"github.com/labstack/gommon/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
const MODEL_PATH = "./ggml-medium.bin"
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
e := echo.New()
|
||
|
|
e.HideBanner = true
|
||
|
|
|
||
|
|
if l, ok := e.Logger.(*log.Logger); ok {
|
||
|
|
l.SetHeader("${time_rfc3339} ${level}")
|
||
|
|
}
|
||
|
|
|
||
|
|
whisperState, err := InitializeWhisperState(MODEL_PATH)
|
||
|
|
if err != nil {
|
||
|
|
e.Logger.Error(err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
e.POST("/v1/audio/transcriptions", func(c echo.Context) error {
|
||
|
|
return transcribe(c, whisperState)
|
||
|
|
})
|
||
|
|
|
||
|
|
e.Logger.Fatal(e.Start(":3000"))
|
||
|
|
}
|