app.py

# Importer pipeline
from transformers import pipeline
# charger le modèle mT5_multilingual_XLSum
summarizer_1= pipeline("summarization", model="csebuetnlp/mT5_multilingual_XLSum")
# charger le modèle bart-large-cnn
summarizer = pipeline(task="summarization", model="facebook/bart-large-cnn")
# Définir une fonction summarize_func avec bart-large-cnn
def summarize_func(input, min_length, max_length):
    output = summarizer(input.strip(),min_length, max_length)
    return output[0]['summary_text']

# Définir une fonction summarize_func avec mt5-multilingual
def summarize_func_1(input, min_length, max_length):
    output = summarizer_1(input.strip(), min_length, max_length)
    return output[0]['summary_text']
    
# Déployer
import gradio as gr
import os
# Création de blocks
demo = gr.Blocks(theme='ibm-software/gradio-carbon-theme')
inputs = [gr.Textbox(label="Text à résumer", lines=6),
          gr.Number(label = 'Longueur Minimal'),
          gr.Number(label = 'Longueur Maximal')]
summarizer1 = gr.Interface(fn=summarize_func,
                    inputs=inputs,
                    outputs=[gr.Textbox(label="Résumé", lines=3)],
                    title="Text summarization avec bart-large-cnn",
                    description="Ce modèle BART-Large-CNN, entraîné sur des données en anglais, permet de résumer automatiquement un texte en générant une version condensée, tout en conservant l’essentiel des informations.\n\nEntrez un texte long en anglais, puis choisissez la longueur minimale et maximale du résumé souhaité."
                   )
inputs2 = [gr.Textbox(label="Texte à résumer", lines=6),
           gr.Number(label='Longueur minimale'),
           gr.Number(label='Longueur maximale')]

summarizer2 = gr.Interface(
    fn=summarize_func_1,
    inputs=inputs2,
    outputs=[gr.Textbox(label="Résumé", lines=3)],
    title="Text summarization avec mT5_multilingual_XLSum",
    description="Ce modèle mT5 multilingue est capable de résumer automatiquement des textes dans plusieurs langues, dont le français, l’anglais et plus encore.\n\nEntrez un texte dans la langue de votre choix et définissez la longueur du résumé souhaitée."
)


with demo:
    gr.TabbedInterface(
        [summarizer1,
         summarizer2],
        ["Summarize avec mT5",
         "Summarize avec bart"],
        )

demo.launch()