Kari commited on
Commit
fd16052
·
1 Parent(s): 4a8465c

Add application file

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -4,10 +4,13 @@ from transformers import pipeline
4
  import os
5
  translator = pipeline("translation", model=os.path.join(os.getcwd(),"test-ml-trained"), max_time=5)
6
 
7
- def en2zh(EnName):
8
- pre = translator(EnName[0], )[0]["translation_text"]
9
- print(pre)
10
- return pre
 
 
 
11
 
12
  interface = gr.Interface(fn=en2zh, inputs="list", outputs="list")
13
  interface.launch()
 
4
  import os
5
  translator = pipeline("translation", model=os.path.join(os.getcwd(),"test-ml-trained"), max_time=5)
6
 
7
+ def en2zh(EnNames):
8
+ pres=[]
9
+ for name in EnNames:
10
+ pre = translator(name, )[0]["translation_text"]
11
+ pres.append(pre)
12
+ print(pres)
13
+ return pres
14
 
15
  interface = gr.Interface(fn=en2zh, inputs="list", outputs="list")
16
  interface.launch()