lang param.

brave-experiments · Jul 31, 2024 · d009232 · d009232
1 parent d349ca8
commit d009232
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/runners/audio_transcriber.py b/src/runners/audio_transcriber.py
@@ -21,7 +21,10 @@ def __init__(self):
     @bentoml.Runnable.method(batchable=False)
     def transcribe_audio(self, audio):
         segments, info = self.model.transcribe(
-            audio, vad_filter=True, vad_parameters=dict(min_silence_duration_ms=500)
+            audio,
+            vad_filter=True,
+            vad_parameters=dict(min_silence_duration_ms=500),
+            language="en",
         )
 
         text = ""