fix default values

app_dialogue.py

@@ -303,7 +303,7 @@ with gr.Blocks(title="IDEFICS", theme=gr.themes.Base()) as demo:
                 top_p = gr.Slider(
                     minimum=0.0,
                     maximum=1.0,
-                    value=0.7,
+                    value=0.95,
                     step=0.1,
                     interactive=True,
                     label="Top P",
@@ -326,7 +326,7 @@ with gr.Blocks(title="IDEFICS", theme=gr.themes.Base()) as demo:
                 )
                 repetition_penalty = gr.Slider(
                     minimum=0.0,
-                    maximum=10.0,
+                    maximum=5.0,
                     value=1.0,
                     step=0.1,
                     interactive=True,
@@ -342,7 +342,7 @@ with gr.Blocks(title="IDEFICS", theme=gr.themes.Base()) as demo:
                 )
                 length_penalty = gr.Slider(
                     minimum=0.0,
-                    maximum=10.0,
+                    maximum=5.0,
                     value=1.0,
                     step=0.1,
                     interactive=True,
@@ -358,9 +358,9 @@ with gr.Blocks(title="IDEFICS", theme=gr.themes.Base()) as demo:
                 )
                 penalty_alpha = gr.Slider(
                     minimum=0.0,
-                    maximum=10.0,
+                    maximum=5.0,
                     value=0.95,
-                    step=1.0,
+                    step=0.05,
                     interactive=True,
                     label="Penalty alpha",
                 )
@@ -435,16 +435,16 @@ with gr.Blocks(title="IDEFICS", theme=gr.themes.Base()) as demo:
     def model_inference(
         user_prompt,
         chat_history,
-        decoding_strategy="greedy",
-        temperature=1.0,
-        no_repeat_ngram_size=0,
-        max_new_tokens=512,
-        min_length=16,
-        repetition_penalty=1.0,
-        length_penalty=1.0,
-        top_k=50,
-        top_p=0.95,
-        penalty_alpha=0.95,
+        decoding_strategy,
+        temperature,
+        no_repeat_ngram_size,
+        max_new_tokens,
+        min_length,
+        repetition_penalty,
+        length_penalty,
+        top_k,
+        top_p,
+        penalty_alpha,
     ):
         global processor, model, tokenizer
         # temperature = 1.0