en-sw

app.py

@@ -19,36 +19,43 @@ from transformers import AutoTokenizer, MarianTokenizer, AutoModel, AutoModelFor
 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_sw = "Helsinki-NLP/opus-mt-en-sw"
 
 tokenizer_es = AutoTokenizer.from_pretrained(model_es)
 tokenizer_fr = AutoTokenizer.from_pretrained(model_fr)
 tokenizer_zh = AutoTokenizer.from_pretrained(model_zh)
+tokenizer_sw = AutoTokenizer.from_pretrained(model_sw)
 
 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_sw = MarianMTModel.from_pretrained(model_sw)
 
 model_es = inseq.load_model("Helsinki-NLP/opus-mt-en-es", "input_x_gradient")
 model_fr = inseq.load_model("Helsinki-NLP/opus-mt-en-fr", "input_x_gradient")
 model_zh = inseq.load_model("Helsinki-NLP/opus-mt-en-zh", "input_x_gradient")
+model_sw = inseq.load_model("Helsinki-NLP/opus-mt-en-sw", "input_x_gradient")
 
 
 dict_models = {
 	'en-es': model_es,
 	'en-fr': model_fr,
 	'en-zh': model_zh,
+	'en-sw': model_sw,
 }
 
 dict_models_tr = {
 	'en-es': model_tr_es,
 	'en-fr': model_tr_fr,
 	'en-zh': model_tr_zh,
+	'en-sw': model_tr_sw,
 }
 
 dict_tokenizer_tr = {
 	'en-es': tokenizer_es,
 	'en-fr': tokenizer_fr,
 	'en-zh': tokenizer_zh,
+	'en-sw': tokenizer_sw,
 }
 
 saliency_examples = [
@@ -227,7 +234,7 @@ with gr.Blocks(js="plotsjs.js") as demo:
 	out_text  = gr.Textbox(label="target text")
 	out_text2  = gr.Textbox(visible=False)
 	var2 = gr.JSON(visible=False)
-	radio_c = gr.Radio(choices=['en-zh', 'en-es', 'en-fr'], value="en-zh", label= '', container=False)
+	radio_c = gr.Radio(choices=['en-zh', 'en-es', 'en-fr', 'en-sw'], value="en-zh", label= '', container=False)
 	btn = gr.Button("Translate")
 	input_mic = gr.HTML(html)