Update step2_whisperapi.py

bug...
Huanshere · Sep 5, 2024 · 67c3cbc · 67c3cbc
1 parent b6e0c3f
commit 67c3cbc
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/core/step2_whisperapi.py b/core/step2_whisperapi.py
@@ -94,13 +94,21 @@ def transcribe_audio(audio_file: str):
     )
 
     audio = open(audio_file, "rb")
-    transcript = client.audio.transcriptions.create(
-        file=audio,
-        model="whisper-1",
-        response_format="verbose_json",
-        timestamp_granularities=["word"],
-        language=WHISPER_LANGUAGE
-    )
+    if WHISPER_LANGUAGE == 'auto':
+        transcript = client.audio.transcriptions.create(
+            file=audio,
+            model="whisper-1",
+            response_format="verbose_json",
+            timestamp_granularities=["word"]
+        )
+    else:
+        transcript = client.audio.transcriptions.create(
+            file=audio,
+            model="whisper-1",
+            response_format="verbose_json",
+            timestamp_granularities=["word"],
+            language=WHISPER_LANGUAGE
+        )
 
     # 保存原始转录文本
     os.makedirs('output/log', exist_ok=True)