You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
967 B
45 lines
967 B
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
// Packages
|
|
whisper "github.com/ggerganov/whisper.cpp/bindings/go/pkg/whisper"
|
|
)
|
|
|
|
func main() {
|
|
flags, err := NewFlags(filepath.Base(os.Args[0]), os.Args[1:])
|
|
if err == flag.ErrHelp {
|
|
os.Exit(0)
|
|
} else if err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
} else if flags.GetModel() == "" {
|
|
fmt.Fprintln(os.Stderr, "Use -model flag to specify which model file to use")
|
|
os.Exit(1)
|
|
} else if flags.NArg() == 0 {
|
|
fmt.Fprintln(os.Stderr, "No input files specified")
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Load model
|
|
model, err := whisper.New(flags.GetModel())
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
}
|
|
defer model.Close()
|
|
|
|
// Process files
|
|
for _, filename := range flags.Args() {
|
|
fmt.Println("Processing", filename)
|
|
if err := Process(model, filename, flags.GetLanguage(), flags.IsSpeedup(), flags.IsTokens()); err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
continue
|
|
}
|
|
}
|
|
}
|