Update app.py

app.py

@@ -36,7 +36,7 @@ def generate_text(model, tokenizer, input_text, max_length, temperature):
     return tokenizer.decode(output[0], skip_special_tokens=True)
 
 # Fonction principale pour générer le texte
-def main(input_text, max_length, temperature, model_name):
+def main(input_text, max_length, temperature, model_name, history):
     """Fonction principale pour générer le texte"""
     global model, tokenizer
     if model is None or tokenizer is None:
@@ -80,10 +80,13 @@ def append_history(history, message):
 demo = gr.Blocks()
 
 with demo:
-    gr.Markdown("# Modèle de Langage")
+    with gr.Row():
+        gr.Markdown("# TrySmallModels Quickly!\n"
+                    "NB : Work Better with Models Under 5B Parameters")
 
     with gr.Row():
         with gr.Column():
+            gr.Markdown("## Sélectionner Un Modèle")
             # Textbox pour entrer le nom du modèle
             model_name_textbox = gr.Textbox(label="Nom du modèle", placeholder="Depot/NomModele Exemple kurakurai/Luth-LFM2-1.2B", interactive=True)
         with gr.Column():
@@ -94,20 +97,22 @@ with demo:
             history = gr.Textbox(label="Historique", lines=5)
     
     with gr.Row():
+        gr.Markdown("## Entrer Un Prompt")
         # Textbox pour entrer le texte d'entrée
-        input_text = gr.Textbox(label="Texte d'entrée")
+        input_text = gr.Textbox(label="Texte d'entrée", lines=5)
     with gr.Row():
         # Sliders pour ajuster la longueur maximale et la température
-        max_length_slider = gr.Slider(50, 500, label="Longueur maximale", value=200)
+        max_length_slider = gr.Slider(50, 4500, label="Longueur maximale", value=200)
         temperature_slider = gr.Slider(0.1, 1.0, label="Température", value=0.7)
 
     with gr.Row():
         # Bouton pour soumettre le texte d'entrée
         submit_button = gr.Button("Soumettre")
 
-    with gr.Row():        
+    with gr.Row():
+        gr.Markdown("## Sortie LLM")
         # Textbox pour afficher le texte généré
-        output_text = gr.Textbox(label="Texte généré")
+        output_text = gr.Textbox(label="Texte généré", lines=5)
 
     # Ajout des interactions pour les boutons
     # Lors du clic sur "Charger Le Modèle", appeler la fonction load_model_button_clicked
@@ -121,7 +126,7 @@ with demo:
     submit_button.click(
         main,
         inputs=[input_text, max_length_slider, temperature_slider, model_name_textbox, history],
-        outputs=[output_text],
+        outputs=[output_text, history],
         queue=False
     )