renad0 commited on
Commit
5ffc201
·
verified ·
1 Parent(s): 7900460

Create nn.py

Browse files
Files changed (1) hide show
  1. nn.py +26 -0
nn.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ !pip install gradio transformers
2
+ import gradio as gr
3
+ from transformers import pipeline
4
+ translator = pipeline("translation", model="facebook/nllb-200-distilled-600M")
5
+ def translate_text(text, src_lang, tgt_lang):
6
+ translation = translator(text, src_lang=src_lang, tgt_lang=tgt_lang)
7
+ return translation[0]['translation_text']
8
+ languages = {
9
+ "English": "eng_Latn",
10
+ "French": "fra_Latn",
11
+ "Arabic": "arb",
12
+ "Spanish": "spa_Latn",
13
+ "German": "deu_Latn",
14
+ "Chinese": "zho_Hans"
15
+ }
16
+ iface = gr.Interface(
17
+ fn=translate_text,
18
+ inputs=[
19
+ gr.Textbox(label="Input Text"),
20
+ gr.Dropdown(choices=list(languages.keys()), label="Source Language", value="English"),
21
+ gr.Dropdown(choices=list(languages.keys()), label="Target Language", value="French")
22
+ ],
23
+ outputs=gr.Textbox(label="Translated Text"),
24
+ title="Translation using NLLB-200"
25
+ )
26
+ iface.launch()