steef68 commited on
Commit
cac496c
·
verified ·
1 Parent(s): 90cb47b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -32,7 +32,7 @@ def format_prompt(message, prompt_type):
32
  prompt = "" + selected_prompt + ""
33
  return prompt
34
 
35
- def generate(prompt_type, message, temperature=0.15, max_new_tokens=2048, top_p=0.95, repetition_penalty=1.0):
36
  formatted_prompt = format_prompt(message, prompt_type)
37
 
38
  generate_kwargs = {
@@ -70,10 +70,10 @@ prompt_selection = gr.Dropdown(label="Type de prompt", choices=list(prompts.keys
70
  additional_inputs = [
71
  prompt_selection,
72
  gr.Textbox(label="Votre message", lines=4),
73
- 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"),
74
  gr.Slider(label="Max new tokens", value=2048, minimum=256, maximum=1024, step=64, interactive=True, info="Le nombre maximum de nouveaux jetons"),
75
  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é"),
76
- 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"),
77
  ]
78
 
79
  # Création de l'interface Gradio
 
32
  prompt = "" + selected_prompt + ""
33
  return prompt
34
 
35
+ def generate(prompt_type, message, temperature=0.75, max_new_tokens=2048, top_p=0.95, repetition_penalty=1.1):
36
  formatted_prompt = format_prompt(message, prompt_type)
37
 
38
  generate_kwargs = {
 
70
  additional_inputs = [
71
  prompt_selection,
72
  gr.Textbox(label="Votre message", lines=4),
73
+ gr.Slider(label="Temperature", value=0.75, minimum=0.0, maximum=1.0, step=0.05, interactive=True, info="Des valeurs plus élevées produisent des résultats plus diversifiés"),
74
  gr.Slider(label="Max new tokens", value=2048, minimum=256, maximum=1024, step=64, interactive=True, info="Le nombre maximum de nouveaux jetons"),
75
  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é"),
76
+ gr.Slider(label="Repetition penalty", value=1.1, minimum=1.0, maximum=2.0, step=0.05, interactive=True, info="Pénaliser les jetons répétés"),
77
  ]
78
 
79
  # Création de l'interface Gradio