Update app.py

app.py

@@ -94,19 +94,19 @@ if __name__ == "__main__":
                 # beam search
                 gr.Markdown(
                     """
-                    Beam Search Parameter
+                    BeamSearchStrategy
                     """)
                 num_beams = gr.Slider(maximum=4, value=1, minimum=1, step=1, label='Number of Beams')
-                length_penalty = gr.Slider(maximum=1, value=0.8, minimum=0, label='Length Penalty')
+                length_penalty = gr.Slider(maximum=1, value=1, minimum=0, label='Length Penalty')
                 no_repeat_ngram_size = gr.Slider(maximum=5, value=3, minimum=1, step=1, label='No Repeat Ngram Size')
             with gr.Column():
                 # base search
                 gr.Markdown(
                     """
-                    Base Search Parameter
+                    BaseStrategy
                     """)
                 temperature = gr.Slider(maximum=1, value=1, minimum=0, label='Temperature')
-                topk = gr.Slider(maximum=8, value=1, minimum=0, step=1, label='Top K')
+                topk = gr.Slider(maximum=40, value=1, minimum=0, step=1, label='Top K')
                 topp = gr.Slider(maximum=1, value=0, minimum=0, label='Top P')
             
         inputs = [model_input, seed, out_seq_length, min_gen_length, sampling_strategy, num_beams, length_penalty, no_repeat_ngram_size, temperature, topk, topp]