en-sw

app.py

@@ -24,36 +24,42 @@ from bertviz_gradio import head_view_mod
 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 = [
@@ -152,7 +158,7 @@ with gr.Blocks(js="plotsjs_bertviz.js") as demo:
         out_text2  = gr.Textbox(visible=False)
         var2 = gr.JSON(visible=False)
         btn = gr.Button("Create sentence.")
-        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)
 
 
       with gr.Column(scale=4):