Spaces:
Running
Running
File size: 910 Bytes
2cabcd4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import gradio as gr
from transformers import BertTokenizerFast
from optimum.onnxruntime import ORTModelForSeq2SeqLM
from translation_pipeline import TranslationPipeline
from translator import Translator
model_id = "indiejoseph/bart-translation-zh-yue-onnx"
tokenizer = BertTokenizerFast.from_pretrained(model_id)
model = ORTModelForSeq2SeqLM.from_pretrained(model_id, use_cache=False)
pipe = TranslationPipeline(model=model, tokenizer=tokenizer)
translator = Translator(pipe, batch_size=2, max_length=1024)
def translate(zh: str):
return translator([zh])[0]
demo = gr.Interface(
fn=translate,
inputs=[
gr.Textbox(label="官話", type="text"),
],
outputs=[
gr.Textbox(label="廣東話", type="text"),
],
examples=[["瞧瞧你说的是人话吗?"], ["余文乐关掉潮店被嘲临走还坑人"]],
)
if __name__ == "__main__":
demo.launch(show_api=False)
|