Modified params

app.py

@@ -37,7 +37,7 @@ def generate(prompt, model_parameters):
         )
 
 
-def submit_input(input_, num_chunks, max_new_tokens, repetition_penalty, top_k, top_p, do_sample, num_beams, temperature):
+def submit_input(input_, num_chunks, max_new_tokens, repetition_penalty, top_k, top_p, do_sample, temperature):
     if input_.strip() == "":
         gr.Warning("Not possible to inference an empty input")
         return None
@@ -75,13 +75,12 @@ def clear():
         None,
         None,
         None,
-        gr.Slider(value=100),
-        gr.Slider(value=1.2),
+        gr.Slider(value=MAX_NEW_TOKENS),
+        gr.Slider(value=1.0),
         gr.Slider(value=50),
-        gr.Slider(value=0.95),
-        gr.Checkbox(value=True),
-        gr.Slider(value=4),
-        gr.Slider(value=0.5),
+        gr.Slider(value=0.99),
+        gr.Checkbox(value=False),
+        gr.Slider(value=0.35),
     )
 
 
@@ -125,7 +124,7 @@ def gradio_app():
                             minimum=1,
                             maximum=6,
                             step=1,
-                            value=4,
+                            value=1,
                             label="Number of chunks"
                         )
                         max_new_tokens = Slider(
@@ -159,13 +158,6 @@ def gradio_app():
                             value=False, 
                             label="Do sample"
                         )
-                        num_beams = Slider(
-                            minimum=1, 
-                            maximum=4, 
-                            step=1,
-                            value=1,
-                            label="Beams"
-                        )
                         temperature = Slider(
                             minimum=0.1, 
                             maximum=1,
@@ -173,7 +165,7 @@ def gradio_app():
                             label="Temperature"
                         )
 
-                        parameters_compontents = [num_chunks, max_new_tokens, repetition_penalty, top_k, top_p, do_sample, num_beams, temperature]
+                        parameters_compontents = [num_chunks, max_new_tokens, repetition_penalty, top_k, top_p, do_sample, temperature]
 
             with gr.Column(variant="panel"):
                 output = Textbox(