Add application file
Browse files
app.py
CHANGED
@@ -1,9 +1,10 @@
|
|
1 |
# Helsinki-NLP/opus-mt-zh-en
|
2 |
import gradio as gr
|
3 |
from transformers import pipeline
|
|
|
|
|
4 |
|
5 |
def en2zh(EnName):
|
6 |
-
translator = pipeline("translation", model=".\\test-ml-trained", max_time=7)
|
7 |
pre = translator(EnName, )[0]["translation_text"]
|
8 |
return pre
|
9 |
|
|
|
1 |
# Helsinki-NLP/opus-mt-zh-en
|
2 |
import gradio as gr
|
3 |
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]["translation_text"]
|
9 |
return pre
|
10 |
|