Update app.py

app.py

@@ -32,7 +32,7 @@ def format_prompt(message, prompt_type):
     prompt = "" + selected_prompt + ""
     return prompt
 
-def generate(prompt_type, message, temperature=0.15, max_new_tokens=1024, top_p=0.95, repetition_penalty=1.0):
+def generate(prompt_type, message, temperature=0.15, max_new_tokens=2048, top_p=0.95, repetition_penalty=1.0):
     formatted_prompt = format_prompt(message, prompt_type)
 
     generate_kwargs = {
@@ -71,7 +71,7 @@ additional_inputs = [
     prompt_selection,
     gr.Textbox(label="Votre message", lines=4),
     gr.Slider(label="Temperature", value=0.15, minimum=0.0, maximum=1.0, step=0.05, interactive=True, info="Des valeurs plus élevées produisent des résultats plus diversifiés"),
-    gr.Slider(label="Max new tokens", value=1024, minimum=256, maximum=1024, step=64, interactive=True, info="Le nombre maximum de nouveaux jetons"),
+    gr.Slider(label="Max new tokens", value=2048, minimum=256, maximum=1024, step=64, interactive=True, info="Le nombre maximum de nouveaux jetons"),
     gr.Slider(label="Top-p (nucleus sampling)", value=0.90, minimum=0.15, maximum=0.95, step=0.05, interactive=True, info="Des valeurs plus élevées échantillonnent davantage de jetons à faible probabilité"),
     gr.Slider(label="Repetition penalty", value=1.2, minimum=1.0, maximum=2.0, step=0.05, interactive=True, info="Pénaliser les jetons répétés"),
 ]