mirror of
https://github.com/xzeldon/whisper-api-server.git
synced 2024-12-28 17:35:44 +00:00
42 lines
854 B
Go
42 lines
854 B
Go
package main
|
|
|
|
import (
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/labstack/echo/v4/middleware"
|
|
"github.com/labstack/gommon/log"
|
|
"github.com/xzeldon/whisper-api-server/internal/api"
|
|
"github.com/xzeldon/whisper-api-server/internal/resources"
|
|
)
|
|
|
|
func main() {
|
|
e := echo.New()
|
|
e.HideBanner = true
|
|
|
|
e.Use(middleware.CORS())
|
|
|
|
if l, ok := e.Logger.(*log.Logger); ok {
|
|
l.SetHeader("${time_rfc3339} ${level}")
|
|
}
|
|
|
|
_, err := resources.GetWhisperDll("1.12.0")
|
|
if err != nil {
|
|
e.Logger.Error(err)
|
|
}
|
|
|
|
model, err := resources.GetModel("ggml-medium.bin")
|
|
if err != nil {
|
|
e.Logger.Error(err)
|
|
}
|
|
|
|
whisperState, err := api.InitializeWhisperState(model)
|
|
if err != nil {
|
|
e.Logger.Error(err)
|
|
}
|
|
|
|
e.POST("/v1/audio/transcriptions", func(c echo.Context) error {
|
|
return api.Transcribe(c, whisperState)
|
|
})
|
|
|
|
e.Logger.Fatal(e.Start("127.0.0.1:3000"))
|
|
}
|