Update app.py

app.py

@@ -60,11 +60,11 @@ class MistralChatBot:
     def __init__(self, system_prompt="Le dialogue suivant est une conversation"):
         self.system_prompt = system_prompt
 
-    def predict(self, user_message, style):
+    def predict(self, user_message, style, temperature):
 
         system_prompt = styles_prompts_dict[style]
 
-        sampling_params = SamplingParams(temperature=0.7, top_p=.95, max_tokens=500, presence_penalty = 2)
+        sampling_params = SamplingParams(temperature=temperature, top_p=.95, max_tokens=500, presence_penalty = 2)
         detailed_prompt = "<|im_start|>system\n" + system_prompt + "<|im_end|>\n<|im_start|>user"""
         detailed_prompt = detailed_prompt + "\n" + user_message + "<|im_end|>\n<|im_start|>assistant\n"
         prompts = [detailed_prompt]
@@ -87,7 +87,7 @@ examples = [
     [
         "Comment faire une bonne madeleine ? Donne-moi une recette détaillée avec les ingrédients.",  # user_message
         "Proust",
-        "Il y avait eu dans le salon de l'hôtel, que j'avais quitté pour venir à la table où il se trouvait M. de Charlus et une foule d'autres personnes, une meublière avec un petit râteau qui remplissait des petits bols en châtaigne les plats pleins de madeleines tout juste sortis du four. Je n'ai jamais vu mieux faire, et je ne saurais vous donner une recette plus exacte de ces délicieuses pâtisseries : prenez deux oeufs, bien battus ; ajoutez une cuillère de beurre fondu, une demi-cuillère de sucre, une poignée de farine de blé tendre mélangeant tous ces ingrédients avec soin pour éviter de former un amas qui se figerait au moment de la cuisson. Faites vos boules puis placez-les sur des plaques allongées faites de papier rouge, étendues sur un plat. Enfin mettez votre plat dans une grande cuisine, ouverture fermée, pendant trente minutes environ ; lorsque les madeleines sont cuites elles doivent être encore souples, légères, non trop brunes et c'est ce qui leur donne cette croûte fine qui est sans doute la partie la plus délicieuse de ces merveilleux petits gâteaux.",  # assistant_message
+        0.7  # temperature
     ]
 ]
 
@@ -126,7 +126,16 @@ list_dropdown = [
 
 
 additional_inputs=[
-    gr.Dropdown(list_dropdown, value="Proust", label="Style", info="Le style prédominant de Bellay")
+    gr.Dropdown(list_dropdown, value="Proust", label="Style", info="Le style prédominant de Bellay"),
+    gr.Slider(
+        label="Température",
+        value=0.7,  # Default value
+        minimum=0.05,
+        maximum=1.0,
+        step=0.05,
+        interactive=True,
+        info="Des valeurs plus élevées donne plus de créativité, mais aussi d'étrangeté",
+    ),
 ]
 
 iface = gr.Interface(