mirror/LocalAI
1
0 Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2026-10-22 06:14:29 -04:00
Code Issues Packages Projects Releases Wiki Activity

whisper: add tests and allow to set upload size (#237)

Browse Source
This commit is contained in:
Ettore Di Giacinto authored and GitHub committed 2023-05-12 10:04:20 +02:00
1 parent 5115b2faa3
commit fd1df4e971
7 files changed
+53 -21
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

pkg/whisper/whisper.go
+5 -1
View File
@@ -28,7 +28,7 @@ func audioToWav(src, dst string) error {
return nil
}
func Transcript(model whisper.Model, audiopath, language string) (string, error) {
func Transcript(model whisper.Model, audiopath, language string, threads uint) (string, error) {
dir, err := os.MkdirTemp("", "whisper")
if err != nil {
@@ -65,8 +65,12 @@ func Transcript(model whisper.Model, audiopath, language string) (string, error)
}
context.SetThreads(threads)
if language != "" {
context.SetLanguage(language)
} else {
context.SetLanguage("auto")
}
if err := context.Process(data, nil); err != nil {