| import pandas as pd | |
| from fastai.text.all import * | |
| try: | |
| import transformers | |
| except ImportError: | |
| os.system('pip install transformers') | |
| import transformers | |
| import gradio as gr | |
| from transformers import pipeline | |
| repo_id = "maviced/mbart-samsum" | |
| summarizer = pipeline('text2text-generation', model=repo_id) | |
| def predict(texto): | |
| return summarizer(texto) | |
| texto1 = 'Sophia: ¿Has terminado de leer ese libro? Luis: Sí, lo terminé anoche. Sophia: ¿Qué te pareció? Luis: Fue increíble. Definitivamente lo recomendaría.' | |
| texto2 = 'Maria: ¿Vas a venir a la fiesta esta noche? Juan: Sí, planeo ir. Maria: ¡Genial! Estoy emocionada de verte allí. Juan: Yo también estoy emocionado de verte. Va a ser una gran noche.' | |
| gr.Interface(fn=predict, inputs="textbox", outputs="textbox", examples=[texto1, texto2]).launch(share=True) |