modify worker encode

chinaboard · Sep 29, 2023 · 22c701f · 22c701f
1 parent 251699c
commit 22c701f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/cmd/worker/main.go b/cmd/worker/main.go
@@ -13,7 +13,7 @@ import (
 )
 
 const (
-	filePath = "/tmp/temp.wav"
+	filePath = "/tmp/temp.m4a"
 )
 
 func main() {

diff --git a/pkg/videodl/videodl.go b/pkg/videodl/videodl.go
@@ -8,7 +8,7 @@ import (
 )
 
 func Download(videoUrl, savePath string) error {
-	cmd := exec.Command("youtube-dl", "-f", "ba", "-x", "--audio-format", "wav", videoUrl, "-o", savePath)
+	cmd := exec.Command("youtube-dl", "-f", "ba", "-x", "--audio-format", "m4a", videoUrl, "-o", savePath)
 
 	var stdout, stderr bytes.Buffer
 	cmd.Stdout = &stdout

diff --git a/pkg/whisper/whisper.go b/pkg/whisper/whisper.go
@@ -11,7 +11,7 @@ func Asr(whisperEndpoint, filePath, language string) (*AsrResp, error) {
 	resp, err := client.R().
 		SetHeader("Accept", "application/json").
 		SetFiles(map[string]string{"audio_file": filePath}).
-		SetQueryString(fmt.Sprintf("task=transcribe&language=%s&encode=false&output=json", language)).
+		SetQueryString(fmt.Sprintf("task=transcribe&language=%s&encode=true&output=json", language)).
 		Post(whisperEndpoint + "/asr")
 
 	if err != nil {
-Original file line number
+Diff line change
@@ Expand Up / @@ -13,7 +13,7 @@ import ( @@
     )
     const (
-    	filePath = "/tmp/temp.wav"
+    	filePath = "/tmp/temp.m4a"
     )
     func main() {
@@ Expand Down @@