Update asr.py

doppeltilde · Jun 13, 2024 · 5576522 · 5576522
1 parent 085c661
commit 5576522
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/routes/api/asr.py b/src/routes/api/asr.py
@@ -16,6 +16,7 @@ async def asr(
     file: UploadFile = File(),
     model_name: str = Query(None),
     word_timestamps: bool = Query(False),
+    beam_size: int = Query(1),
     language: str = Query(
         None,
         enum=LANGUAGE_CODES,
@@ -32,7 +33,7 @@ async def asr(
 
         segments, info = model.transcribe(
             audio_file,
-            beam_size=5,
+            beam_size=beam_size,
             word_timestamps=True if word_timestamps else False,
             language=language,
         )