From 5576522d947ea2a3bf353e32f125199ec324cc49 Mon Sep 17 00:00:00 2001
From: "Jona T. Feucht" <14951074+jonafeucht@users.noreply.github.com>
Date: Thu, 13 Jun 2024 18:45:54 +0200
Subject: [PATCH] Update asr.py

---
 src/routes/api/asr.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/routes/api/asr.py b/src/routes/api/asr.py
index b02e905..366c90e 100644
--- 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,
         )