File size: 844 Bytes
6bd6f17
e62b25d
6bd6f17
 
 
 
 
 
 
d8f9471
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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)