lauragordo commited on
Commit
e3e07e5
1 Parent(s): 2901455

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -10,14 +10,12 @@ except ImportError:
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)
 
10
  import gradio as gr
11
  from transformers import pipeline
12
 
13
+ repo_id = "lauragordo/mbart-traduct"
14
+ traduct = pipeline('text2text-generation', model=repo_id)
15
 
16
  def predict(texto):
17
+ return traduct(texto)
18
 
 
 
19
 
20
 
21
+ gr.Interface(fn=predict, inputs="textbox", outputs="textbox", examples=["When is your birthday", "i like flowers"]).launch(share=True)