joinfv commited on
Commit
2642f0e
·
verified ·
1 Parent(s): 05adec7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,13 +8,13 @@ from transformers import pipeline
8
  #load model
9
  pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-en-pt")
10
 
11
- def prever(text):
12
  return pipe(text)[0]["translation_text"]
13
 
14
  title = "Tradutor inglês para Brasileires"
15
 
16
  iface = gr.Interface(
17
- fn=prever,
18
  inputs=[gr.Textbox(label="text", lines=3)],
19
  outputs='text',
20
  title=title,
 
8
  #load model
9
  pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-en-pt")
10
 
11
+ def predict(text):
12
  return pipe(text)[0]["translation_text"]
13
 
14
  title = "Tradutor inglês para Brasileires"
15
 
16
  iface = gr.Interface(
17
+ fn=predict,
18
  inputs=[gr.Textbox(label="text", lines=3)],
19
  outputs='text',
20
  title=title,