ar

app.py

@@ -23,31 +23,37 @@ metadata_all = {}
 model_es = "Helsinki-NLP/opus-mt-en-es"
 model_fr = "Helsinki-NLP/opus-mt-en-fr"
 model_zh = "Helsinki-NLP/opus-mt-en-zh"
+model_ar = "Helsinki-NLP/opus-mt-en-ar"
 
 tokenizer_es = AutoTokenizer.from_pretrained(model_es)
 tokenizer_fr = AutoTokenizer.from_pretrained(model_fr)
 tokenizer_zh = AutoTokenizer.from_pretrained(model_zh)
+tokenizer_ar = AutoTokenizer.from_pretrained(model_ar)
 
 model_tr_es = MarianMTModel.from_pretrained(model_es)
 model_tr_fr = MarianMTModel.from_pretrained(model_fr)
 model_tr_zh = MarianMTModel.from_pretrained(model_zh)
+model_tr_ar = MarianMTModel.from_pretrained(model_ar)
 
 dict_models = {
 	'en-es': model_es,
 	'en-fr': model_fr,
 	'en-zh': model_zh,
+	'en-ar': model_ar,
 }
 
 dict_models_tr = {
 	'en-es': model_tr_es,
 	'en-fr': model_tr_fr,
 	'en-zh': model_tr_zh,
+	'en-ar': model_tr_ar,
 }
 
 dict_tokenizer_tr = {
 	'en-es': tokenizer_es,
 	'en-fr': tokenizer_fr,
 	'en-zh': tokenizer_zh,
+	'en-ar': tokenizer_ar,
 }
 
 from faiss import write_index, read_index
@@ -687,7 +693,7 @@ with gr.Blocks(js="plotsjs.js") as demo:
 	""")
 	with gr.Row():
 		with gr.Column(scale=1):
-			model_radio_c = gr.Radio(choices=['en-es', 'en-zh', 'en-fr'], value="en-es", label= '', container=False)
+			model_radio_c = gr.Radio(choices=['en-es', 'en-zh', 'en-fr', 'en-ar'], value="en-es", label= '', container=False)
 
 		with gr.Column(scale=2):
 			gr.Markdown(