Spaces:
Sleeping
Sleeping
File size: 3,055 Bytes
d1779df 45182e9 80f9484 2129b08 80f9484 2129b08 80f9484 2129b08 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 | # Importer pipeline
from transformers import pipeline
# charger le modèle bart-large-cnn
summarizer = pipeline(task="summarization", model="facebook/bart-large-cnn")
# charger le modèle bart-large-cnn
summarizer = pipeline(task="summarization", model="facebook/bart-large-cnn")
# charger le modèle mT5_multilingual_XLSum
summarizer_1= pipeline("summarization", model="csebuetnlp/mT5_multilingual_XLSum")
# 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']
# charger le modèle mT5_multilingual_XLSum
summarizer_1= pipeline("summarization", model="csebuetnlp/mT5_multilingual_XLSum")
# 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é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='NoCrypt/miku')
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="Résumer n'importe quel texte avec bart-large-cnn"
)
inputs1 = [gr.Textbox(label="Text à résumer", lines=6),
gr.Number(label = 'Longueur Minimal'),
gr.Number(label = 'Longueur Maximal')]
summarizer2 = gr.Interface(fn=summarize_func_1,
inputs=inputs1,
outputs=[gr.Textbox(label="Result", lines=3)],
title="Text summarization avec mT5_multilingual_XLSum",
description="Résumer n'importe quel texte mT5_multilingual_XLSum"
)
with demo:
gr.TabbedInterface(
[summarizer1,
summarizer2],
["Summarize avec mT5",
"Summarize avec bart"],
)
demo.launch() |