en-sw

app.py

@@ -21,10 +21,13 @@ from functools import partial
 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"
+
 
 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")
 
 
 def saliency_contrastive(input, ref, cont, inseq_model):
@@ -89,7 +92,7 @@ dict_models = {
 	'en-zh': model_zh,
 	'en-fr': model_fr,
 	'en-es': model_es,
-	# 'en-sw': model_sw
+	'en-sw': model_sw
 }
 
 saliency_examples = [
@@ -144,7 +147,7 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
 		gr.Examples(contrastive_examples,[text_input_c, text_input_ref, text_input_cont])
 		text_output_c = gr.HTML()
 
-		radio_c = gr.Radio(choices=['en-zh', 'en-es', 'en-fr'], value="en-fr", label= '', container=False)
+		radio_c = gr.Radio(choices=['en-zh', 'en-es', 'en-fr', 'en-sw'], value="en-fr", label= '', container=False)
 		text_button_c = gr.Button("Translate")
 
 	text_button.click(fn=saliency_inseq, inputs=[text_input, radio], outputs=[text_output, text_output_box])