Update fn.py

fn.py

@@ -4,6 +4,7 @@ model = None
 model_size = None
 initial_prompt = None
 language = 'ja'
+transcribe_kwargs = {}
 
 def load_model(_model_size):
     global model_size, model
@@ -22,8 +23,12 @@ def set_prompt(prompt, _language = None):
     if _language:
         language = _language
 
+def set_transcribe_kwargs(args):
+    global transcribe_kwargs
+    transcribe_kwargs = args
+
 def speech_to_text(audio_file, _model_size = None):
-    global model_size, model
+    global model_size, model, transcribe_kwargs
 
     load_model(_model_size)
 
@@ -34,6 +39,7 @@ def speech_to_text(audio_file, _model_size = None):
         beam_size=5,
         vad_filter=True,
         without_timestamps=False,
+        **transcribe_kwargs,
     )
 
     text_only = ''