dofbi commited on
Commit
dc03a05
·
1 Parent(s): 9741eb6
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -25,7 +25,7 @@ def generate_response(user_input, system_prompt, max_new_tokens=150, temperature
25
  # Gérer le cas où le prompt système est vide
26
  if not system_prompt.strip():
27
  messages = [
28
- {"role": "user", "content": "Translate to Wolof: Bassirou Diomaye Faye is the new Senegalese president. He is 44 years old"}
29
  ]
30
  else:
31
  messages = [
@@ -78,6 +78,9 @@ with gr.Blocks() as iface:
78
  # Slider pour choisir le nombre maximal de tokens
79
  max_tokens_slider = gr.Slider(50, 500, value=150, label="Nombre max de tokens")
80
 
 
 
 
81
  # Bouton pour générer la réponse
82
  generate_button = gr.Button("Générer une réponse")
83
 
@@ -88,7 +91,7 @@ with gr.Blocks() as iface:
88
  # Connecter le bouton à la fonction de génération
89
  generate_button.click(
90
  fn=generate_response,
91
- inputs=[user_input, system_prompt_textbox, max_tokens_slider],
92
  outputs=output
93
  )
94
 
 
25
  # Gérer le cas où le prompt système est vide
26
  if not system_prompt.strip():
27
  messages = [
28
+ {"role": "user", "content": user_input}
29
  ]
30
  else:
31
  messages = [
 
78
  # Slider pour choisir le nombre maximal de tokens
79
  max_tokens_slider = gr.Slider(50, 500, value=150, label="Nombre max de tokens")
80
 
81
+ # Slider pour ajuster la température
82
+ temperature_slider = gr.Slider(0.1, 1.5, value=0.7, step=0.1, label="Température (créativité)")
83
+
84
  # Bouton pour générer la réponse
85
  generate_button = gr.Button("Générer une réponse")
86
 
 
91
  # Connecter le bouton à la fonction de génération
92
  generate_button.click(
93
  fn=generate_response,
94
+ inputs=[user_input, system_prompt_textbox, max_tokens_slider, temperature_slider],
95
  outputs=output
96
  )
97