svjack's picture
Upload app.py
b38e725
from easynmt import EasyNMT
import os
#zh_en_naive_model = EasyNMT("m2m_100_418M")
#p00 = os.path.join(model_path, "zh_en_m2m")
p00 = "zh_en_m2m"
assert os.path.exists(p00)
zh_en_naive_model = EasyNMT(p00)
zh_en_naive_model.translate(["宁波在哪?"], source_lang="zh", target_lang = "en")
'''
en_question = zh_en_naive_model.translate([zh_question], source_lang="zh", target_lang = "en")[0]
'''
import gradio as gr
example_sample = [
"宁波在哪个省份?",
"美国的通货是什么?",
]
def demo_func(zh_question):
assert type(zh_question) == type("")
en_question = zh_en_naive_model.translate([zh_question], source_lang="zh", target_lang = "en")[0]
return {
"English Question": en_question
}
demo = gr.Interface(
fn=demo_func,
inputs=gr.Text(label = "Chinese Question"),
outputs="json",
title=f"Translate Chinese into English 🐱 demonstration",
examples=example_sample if example_sample else None,
cache_examples = False
)
demo.launch(server_name=None, server_port=None)