File size: 995 Bytes
7900460 |
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 |
!pip install gradio transformers
import gradio as gr
from transformers import pipeline
translator = pipeline("translation", model="facebook/nllb-200-distilled-600M")
def translate_text(text, src_lang, tgt_lang):
translation = translator(text, src_lang=src_lang, tgt_lang=tgt_lang)
return translation[0]['translation_text']
languages = {
"English": "eng_Latn",
"French": "fra_Latn",
"Arabic": "arb",
"Spanish": "spa_Latn",
"German": "deu_Latn",
"Chinese": "zho_Hans"
}
iface = gr.Interface(
fn=translate_text,
inputs=[
gr.Textbox(label="Input Text"),
gr.Dropdown(choices=list(languages.keys()), label="Source Language", value="English"),
gr.Dropdown(choices=list(languages.keys()), label="Target Language", value="French")
],
outputs=gr.Textbox(label="Translated Text"),
title="Translation using NLLB-200"
)
iface.launch()
|