|
from easynmt import EasyNMT |
|
import os |
|
|
|
|
|
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) |
|
|