Update app.py

app.py

@@ -22,11 +22,6 @@ else:
 def generate_text(*args):
     if args[0] == "":
         return "Please input text"
-    args[1] = 0.6 if args[1] == 0 else args[1]
-    args[3] = 5 if args[3] == 0 else args[3]
-    args[4] = 0.5 if args[4] == 0 else args[4]
-    args[5] = 50 if args[5] == 0 else args[5]
-
     generation_config = GenerationConfig(
     penalty_alpha=args[1],
     do_sample=args[2],
@@ -73,17 +68,17 @@ with gr.Blocks() as demo:
         inp_text = gr.Textbox(label="Input Text")
         inp = gr.Audio(type="numpy")
         out_text_predict = gr.Textbox(label="Recognized Speech")
-        button = gr.Button("Recognize Speech" , size="sm")
+        button = gr.Button("Recognize Speech" , size="lg")
         button.click(recognize_speech, inp, out_text_predict)
     with gr.Row():
         with gr.Row():
-            penalty_alpha_slider = gr.Slider(minimum=0, maximum=1, step=0.1, label="penalty alpha")
-            do_sample_checkbox = gr.Checkbox(label="do sample")
-            top_k_slider = gr.Slider(minimum=0, maximum=10, step=1, label="top k")
+            penalty_alpha_slider = gr.Slider(minimum=0, maximum=1, step=0.1, label="penalty alpha",value=0.6)
+            do_sample_checkbox = gr.Checkbox(label="do sample",value=True)
+            top_k_slider = gr.Slider(minimum=0, maximum=10, step=1, label="top k", value=5)
         with gr.Row():
-            temperature_slider = gr.Slider(minimum=0, maximum=1, step=0.1, label="temperature")
-            repetition_penalty_slider = gr.Slider(minimum=0, maximum=2, step=0.1, label="repetition penalty")
-            max_new_tokens_slider = gr.Slider(minimum=0, maximum=100, step=1, label="max new tokens")
+            temperature_slider = gr.Slider(minimum=0, maximum=1, step=0.1, label="temperature",value=0.5)
+            repetition_penalty_slider = gr.Slider(minimum=0, maximum=2, step=0.1, label="repetition penalty",value=1.0)
+            max_new_tokens_slider = gr.Slider(minimum=0, maximum=100, step=1, label="max new tokens",value=30)
     with gr.Row():
         out_answer = gr.Textbox(label="Answer")
         button_answer = gr.Button("Answer")