Update app.py

app.py

@@ -5,14 +5,29 @@ import torch
 from datetime import datetime
 import os
 
-# Model description
+
+Title = """# Welcome to 🌟Tonic's 🌠Lucie-7B-Instruct Demo"""
+
 description = """
-# 🇫🇷 Lucie-7B-Instruct
+🌠Lucie-7B-Instruct is a fine-tuned version of [Lucie-7B](https://huggingface.co/OpenLLM-France/Lucie-7B), an open-source, multilingual causal language model created by OpenLLM-France.
 
-Lucie is a French language model based on Mistral-7B, fine-tuned on French data and instructions. 
-This demo allows you to interact with the model and adjust various generation parameters.
+🌠Lucie-7B-Instruct is fine-tuned on synthetic instructions produced by ChatGPT and Gemma and a small set of customized prompts about OpenLLM and Lucie.
 """
 
+training = """
+## Training details
+
+### Training data
+
+Lucie-7B-Instruct is trained on the following datasets:
+* [Alpaca-cleaned](https://huggingface.co/datasets/yahma/alpaca-cleaned) (English; 51604 samples)
+* [Alpaca-cleaned-fr](https://huggingface.co/datasets/cmh/alpaca_data_cleaned_fr_52k) (French; 51655 samples)
+* [Magpie-Gemma](https://huggingface.co/datasets/Magpie-Align/Magpie-Gemma2-Pro-200K-Filtered) (English; 195167 samples)
+* [Wildchat](https://huggingface.co/datasets/allenai/WildChat-1M) (French subset; 26436 samples)
+* Hard-coded prompts concerning OpenLLM and Lucie (based on [allenai/tulu-3-hard-coded-10x](https://huggingface.co/datasets/allenai/tulu-3-hard-coded-10x))
+    * French: openllm_french.jsonl (24x10 samples)
+    * English: openllm_english.jsonl (24x10 samples)"""
+
 join_us = """
 ## Join us:
 🌟TeamTonic🌟 is always making cool demos! Join our active builder's 🛠️community 👻 
@@ -102,7 +117,11 @@ with gr.Blocks() as demo:
                     **Padding Token:** {tokenizer.pad_token}
                     **EOS Token:** {tokenizer.eos_token}
                     """)
-        with gr.Row():
+
+        with gr.Collumn():
+            gr.Markdown(training)
+    
+    with gr.Row():
             gr.Markdown(join_us)
     
     with gr.Row():
@@ -116,54 +135,54 @@ with gr.Blocks() as demo:
             
             # User prompt
             user_prompt = gr.Textbox(
-                label="Votre message",
+                label="🗣️Votre message",
                 placeholder="Entrez votre texte ici...",
                 lines=5
             )
             
-            with gr.Accordion("Paramètres avancés", open=False):
+            with gr.Accordion("🧪Paramètres avancés", open=False):
                 temperature = gr.Slider(
                     minimum=0.1,
                     maximum=2.0,
                     value=0.7,
                     step=0.1,
-                    label="Temperature"
+                    label="🌡️Temperature"
                 )
                 max_new_tokens = gr.Slider(
                     minimum=1,
                     maximum=2048,
                     value=512,
                     step=1,
-                    label="Longueur maximale"
+                    label="💶Longueur maximale"
                 )
                 top_p = gr.Slider(
                     minimum=0.1,
                     maximum=1.0,
                     value=0.9,
                     step=0.1,
-                    label="Top-p"
+                    label="🏅Top-p"
                 )
                 top_k = gr.Slider(
                     minimum=1,
                     maximum=100,
                     value=50,
                     step=1,
-                    label="Top-k"
+                    label="🏆Top-k"
                 )
                 repetition_penalty = gr.Slider(
                     minimum=1.0,
                     maximum=2.0,
                     value=1.2,
                     step=0.1,
-                    label="Pénalité de répétition"
+                    label="🦜Pénalité de répétition"
                 )
             
-            generate_btn = gr.Button("Générer")
+            generate_btn = gr.Button("🌠Générer")
         
         with gr.Column():
             # Output component
             output = gr.Textbox(
-                label="Réponse de Lucie",
+                label="🌠Lucie",
                 lines=10
             )
     
@@ -227,7 +246,7 @@ with gr.Blocks() as demo:
             top_k
         ],
         outputs=output,
-        label="Exemples de prompts avec paramètres optimisés"
+        label="Exemples"
     )
     
     # Set up the generation event