Spaces:
Runtime error
Runtime error
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) |