Update app.py

app.py

@@ -1,5 +1,4 @@
 import torch
-import numpy
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 import gradio as gr
 
@@ -8,113 +7,81 @@ model_name = "Moustapha91/bart_large_poetique-v02"
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-# Fonction de génération
-def generate_poem(prompt, temperature, top_k, top_p, num_beams, repetition_penalty, max_length):
-    """
-    Génère un texte poétique à partir d'un prompt et de paramètres personnalisés.
-    """
-    # Tokenisation du prompt
+# Fonction de génération ajustée
+def generate_summary(input_text):
     inputs = tokenizer(
-        prompt,
+        input_text,
         padding="max_length",
-        max_length=512,
+        max_length=512,  # Ajustez si nécessaire
         return_tensors="pt",
         truncation=True,
     )
     input_ids = inputs.input_ids.to(model.device)
     attention_mask = inputs.attention_mask.to(model.device)
 
-    # Génération avec les paramètres spécifiés par l'utilisateur
+    # Génération avec des paramètres personnalisés
     outputs = model.generate(
         input_ids=input_ids,
         attention_mask=attention_mask,
-        max_length=max_length,
-        temperature=temperature,
-        top_k=top_k,
-        top_p=top_p,
-        num_beams=num_beams,
-        repetition_penalty=repetition_penalty,
+        max_length=512,
+        num_beams=4,
+        temperature=1.0,
         early_stopping=True,
-        no_repeat_ngram_size=3,  # Empêche les répétitions courtes
+        repetition_penalty=6.0,
+        no_repeat_ngram_size=5,
     )
-    # Décodage du texte généré
-    output_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return output_text
+    output_str = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return output_str
 
-# Interface Gradio
-def gradio_interface():
-    # Définir les composants Gradio
-    title = "✨ Générateur de Texte Poétique"
-    description = (
-        "Cette interface interactive vous permet de générer des textes poétiques "
-        "en entrant un 'prompt' et en ajustant les paramètres de génération."
-    )
-    examples = [
-        # Chaque exemple correspond à [prompt, temperature, top_k, top_p, num_beams, repetition_penalty, max_length]
-        ["Dans la clarté d'une nuit étoilée, un fleuve murmure des secrets oubliés.", 1.0, 50, 0.9, 4, 1.5, 150],
-        ["L'amour est un feu doux, éclairant les âmes perdues dans l'ombre.", 1.2, 40, 0.85, 5, 2.0, 100],
-        ["Un enfant rêve d'unifier un continent par des chants de liberté.", 0.8, 30, 0.95, 6, 1.8, 200],
-    ]
+# Liste d'exemples enrichis
+examples = [
+    "L'amour, cet élan mystérieux qui unit les âmes, est une lumière douce dans les ténèbres du quotidien.",
+    "Les tirailleurs sénégalais, ces soldats venus d'Afrique, ont marqué l'histoire par leur courage et leur sacrifice. Recrutés à partir de 1857 sous l'autorité coloniale française, ils provenaient de nombreux pays d'Afrique de l'Ouest.",
+    "Il était une fois un garçon plein de rêves et d'espoir. Porté par une vision audacieuse, il rêvait d'unifier toute l'Afrique sous un même idéal de paix et de solidarité.",
+    "L'eau, source de vie, est l'élément essentiel qui relie tous les êtres vivants. Elle coule dans les rivières, traverse les montagnes et nourrit la terre.",
+    "Dans un petit village au cœur de l'Afrique, une vieille femme racontait les légendes des ancêtres à la lumière des étoiles, transmettant ainsi leur sagesse aux générations futures."
+]
 
-    interface = gr.Interface(
-        fn=generate_poem,
-        inputs=[
-            gr.Textbox(
-                label="Prompt",
-                placeholder="Entrez votre prompt ici...",
-                lines=2,
-            ),
-            gr.Slider(
-                label="Température (Créativité)",
-                minimum=0.5,
-                maximum=1.5,
-                value=1.0,
-                step=0.1,
-            ),
-            gr.Slider(
-                label="Top-k (Mots possibles)",
-                minimum=0,
-                maximum=100,
-                value=50,
-                step=1,
-            ),
-            gr.Slider(
-                label="Top-p (Nucleus Sampling)",
-                minimum=0.5,
-                maximum=1.0,
-                value=0.9,
-                step=0.1,
-            ),
-            gr.Slider(
-                label="Nombre de faisceaux (Beam Search)",
-                minimum=1,
-                maximum=10,
-                value=4,
-                step=1,
-            ),
-            gr.Slider(
-                label="Pénalité de répétition",
-                minimum=1.0,
-                maximum=5.0,
-                value=1.5,
-                step=0.1,
-            ),
-            gr.Slider(
-                label="Longueur maximale",
-                minimum=50,
-                maximum=512,
-                value=150,
-                step=10,
-            ),
-        ],
-        outputs=gr.Textbox(label="Texte Poétique Généré", lines=10),
-        title=title,
-        description=description,
-        examples=examples,
-    )
-    return interface
+# Personnalisation de la description et de l'interface
+title = "📝 Générateur de Texte Poétique et Résumé"
+description = """
+Bienvenue dans notre application interactive de génération de texte poétique ou résumé ! 🖋️  
+Utilisez ce modèle pré-entraîné pour transformer des idées en textes élégants et expressifs.  
+Ajoutez un paragraphe dans la zone ci-dessous et voyez la magie opérer !
+"""
+about = """
+### À propos
+Ce projet utilise le modèle **BART** de Facebook, connu pour ses capacités avancées en résumé et génération de texte.  
+Créé pour inspirer, éduquer et honorer l'héritage de la littérature et des récits africains.  
+Vous pouvez tester avec les exemples ci-dessus ou entrer vos propres idées.
+
+---
+
+📚 **Exemples suggérés :**
+- Un texte sur l'amour.  
+- Une introduction sur les tirailleurs sénégalais.  
+- Un récit d'unité et d'espoir en Afrique.
+"""
+
+# Interface Gradio améliorée
+interface = gr.Interface(
+    fn=generate_summary,
+    inputs=gr.Textbox(
+        lines=10,
+        placeholder="Entrez un texte à transformer en poème ou résumé ici...",
+        label="🖊️ Texte à transformer"
+    ),
+    outputs=gr.Textbox(
+        lines=10,
+        label="✨ Texte généré"
+    ),
+    examples=examples,
+    title=title,
+    description=description,
+    theme="compact",  # Thème compact et moderne
+    article=about,  # Section 'À propos'
+    allow_flagging="never",  # Désactive le flagging si inutile
+)
 
-# Lancer l'application Gradio
-if __name__ == "__main__":
-    interface = gradio_interface()
-    interface.launch()
+# Lancer l'application
+interface.launch()