|
import gradio as gr |
|
from transformers import pipeline |
|
|
|
pipe = pipeline(task="translation", model="facebook/nllb-200-distilled-600M") |
|
|
|
|
|
def greet(text,src_lang,tgt_lang): |
|
|
|
result=pipe(text,src_lang=src_lang ,tgt_lang=tgt_lang) |
|
return result[0]["translation_text"] |
|
|
|
text = gr.Textbox(lines=5,label="输入文本",placeholder="请输入文本") |
|
src_lang = gr.Dropdown(label="源语言",choices=["afr_Latn", |
|
"amh_Ethi", |
|
"arb_Arab", |
|
"asm_Beng", |
|
"ast_Latn", |
|
"azj_Latn", |
|
"bel_Cyrl", |
|
"ben_Beng", |
|
"bos_Latn", |
|
"bul_Cyrl", |
|
"cat_Latn", |
|
"ceb_Latn", |
|
"ces_Latn", |
|
"ckb_Arab", |
|
"cym_Latn", |
|
"dan_Latn", |
|
"deu_Latn", |
|
"ell_Grek", |
|
"eng_Latn", |
|
"est_Latn", |
|
"fin_Latn", |
|
"fra_Latn", |
|
"fuv_Latn", |
|
"gle_Latn", |
|
"glg_Latn", |
|
"guj_Gujr", |
|
"hau_Latn", |
|
"heb_Hebr", |
|
"hin_Deva", |
|
"hrv_Latn", |
|
"hun_Latn", |
|
"hye_Armn", |
|
"ibo_Latn", |
|
"ind_Latn", |
|
"isl_Latn", |
|
"ita_Latn", |
|
"jav_Latn", |
|
"jpn_Jpan", |
|
"kam_Latn", |
|
"kan_Knda", |
|
"kat_Geor", |
|
"kaz_Cyrl", |
|
"khm_Khmr", |
|
"kir_Cyrl", |
|
"kor_Hang", |
|
"lao_Laoo", |
|
"lij_Latn", |
|
"lim_Latn", |
|
"lin_Latn", |
|
"lit_Latn", |
|
"ltz_Latn", |
|
"lug_Latn", |
|
"luo_Latn", |
|
"lvs_Latn", |
|
"mal_Mlym", |
|
"mar_Deva", |
|
"mkd_Cyrl", |
|
"mlt_Latn", |
|
"khk_Cyrl", |
|
"mri_Latn", |
|
"mya_Mymr", |
|
"nld_Latn", |
|
"nob_Latn", |
|
"npi_Deva", |
|
"nso_Latn", |
|
"nya_Latn", |
|
"oci_Latn", |
|
"gaz_Latn", |
|
"ory_Orya", |
|
"pan_Guru", |
|
"pes_Arab", |
|
"pol_Latn", |
|
"por_Latn", |
|
"pbt_Arab", |
|
"ron_Latn", |
|
"rus_Cyrl", |
|
"slk_Latn", |
|
"sna_Latn", |
|
"snd_Arab", |
|
"som_Latn", |
|
"spa_Latn", |
|
"srp_Cyrl", |
|
"swe_Latn", |
|
"swh_Latn", |
|
"tam_Taml", |
|
"tel_Telu", |
|
"tgk_Cyrl", |
|
"tgl_Latn", |
|
"tha_Thai", |
|
"tur_Latn", |
|
"ukr_Cyrl", |
|
"umb_Latn", |
|
"urd_Arab", |
|
"uzn_Latn", |
|
"vie_Latn", |
|
"wol_Latn", |
|
"xho_Latn", |
|
"yor_Latn", |
|
"zho_Hans", |
|
"zho_Hant", |
|
"zsm_Latn", |
|
"zul_Latn"]) |
|
tgt_lang = gr.Dropdown(label="目标语言",choices=["afr_Latn", |
|
"amh_Ethi", |
|
"arb_Arab", |
|
"asm_Beng", |
|
"ast_Latn", |
|
"azj_Latn", |
|
"bel_Cyrl", |
|
"ben_Beng", |
|
"bos_Latn", |
|
"bul_Cyrl", |
|
"cat_Latn", |
|
"ceb_Latn", |
|
"ces_Latn", |
|
"ckb_Arab", |
|
"cym_Latn", |
|
"dan_Latn", |
|
"deu_Latn", |
|
"ell_Grek", |
|
"eng_Latn", |
|
"est_Latn", |
|
"fin_Latn", |
|
"fra_Latn", |
|
"fuv_Latn", |
|
"gle_Latn", |
|
"glg_Latn", |
|
"guj_Gujr", |
|
"hau_Latn", |
|
"heb_Hebr", |
|
"hin_Deva", |
|
"hrv_Latn", |
|
"hun_Latn", |
|
"hye_Armn", |
|
"ibo_Latn", |
|
"ind_Latn", |
|
"isl_Latn", |
|
"ita_Latn", |
|
"jav_Latn", |
|
"jpn_Jpan", |
|
"kam_Latn", |
|
"kan_Knda", |
|
"kat_Geor", |
|
"kaz_Cyrl", |
|
"khm_Khmr", |
|
"kir_Cyrl", |
|
"kor_Hang", |
|
"lao_Laoo", |
|
"lij_Latn", |
|
"lim_Latn", |
|
"lin_Latn", |
|
"lit_Latn", |
|
"ltz_Latn", |
|
"lug_Latn", |
|
"luo_Latn", |
|
"lvs_Latn", |
|
"mal_Mlym", |
|
"mar_Deva", |
|
"mkd_Cyrl", |
|
"mlt_Latn", |
|
"khk_Cyrl", |
|
"mri_Latn", |
|
"mya_Mymr", |
|
"nld_Latn", |
|
"nob_Latn", |
|
"npi_Deva", |
|
"nso_Latn", |
|
"nya_Latn", |
|
"oci_Latn", |
|
"gaz_Latn", |
|
"ory_Orya", |
|
"pan_Guru", |
|
"pes_Arab", |
|
"pol_Latn", |
|
"por_Latn", |
|
"pbt_Arab", |
|
"ron_Latn", |
|
"rus_Cyrl", |
|
"slk_Latn", |
|
"sna_Latn", |
|
"snd_Arab", |
|
"som_Latn", |
|
"spa_Latn", |
|
"srp_Cyrl", |
|
"swe_Latn", |
|
"swh_Latn", |
|
"tam_Taml", |
|
"tel_Telu", |
|
"tgk_Cyrl", |
|
"tgl_Latn", |
|
"tha_Thai", |
|
"tur_Latn", |
|
"ukr_Cyrl", |
|
"umb_Latn", |
|
"urd_Arab", |
|
"uzn_Latn", |
|
"vie_Latn", |
|
"wol_Latn", |
|
"xho_Latn", |
|
"yor_Latn", |
|
"zho_Hans", |
|
"zho_Hant", |
|
"zsm_Latn", |
|
"zul_Latn"]) |
|
output_text = gr.Textbox(lines=5,label="翻译文本",placeholder="翻译文本") |
|
demo=gr.Interface(fn=greet, inputs=[text,src_lang,tgt_lang], outputs=output_text) |
|
demo.launch(share=True) |
