lauragordo commited on
Commit
6b65072
1 Parent(s): be975aa

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -0
app.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pandas as pd
2
+ from fastai.text.all import *
3
+
4
+ try:
5
+ import transformers
6
+ except ImportError:
7
+ os.system('pip install transformers')
8
+ import transformers
9
+
10
+ import gradio as gr
11
+ from transformers import pipeline
12
+
13
+ repo_id = "maviced/mbart-samsum"
14
+ summarizer = pipeline('text2text-generation', model=repo_id)
15
+
16
+ def predict(texto):
17
+ return summarizer(texto)
18
+
19
+ 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.'
20
+ 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.'
21
+
22
+
23
+ gr.Interface(fn=predict, inputs="textbox", outputs="textbox", examples=[texto1, texto2]).launch(share=True)