| !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() | |