adding dropdown for model selection

app.py

@@ -111,6 +111,8 @@ def opus_trans(article, target_language):
         target_lang = "en"
     elif target_language == "Chinese":
         target_lang = "zh"
+    elif target_language == "Spanish":
+        target_lang = "es"
 
     if result_lang != target_lang:
         task_name = f"translation_{result_lang}_to_{target_lang}"
@@ -129,9 +131,12 @@ myTheme = myTheme()
 
 with gr.Blocks(theme=myTheme) as demo:
     article = gr.Textbox(label="Article")
+    toolkit_select = gr.Dropdown(["OPUS", "NLLB"], label="Select Translation Model")
     lang_select = gr.Radio(["English", "Chinese"], label="Select Desired Language")
     result = gr.Textbox(label="Translated Result")
     trans_btn = gr.Button("Translate")
-    trans_btn.click(fn=opus_trans, inputs=[article, lang_select], outputs=result)
-
+    if toolkit_select.select == "OPUS":
+        trans_btn.click(fn=opus_trans, inputs=[article, lang_select], outputs=result)
+    elif toolkit_select.select == "NLLB":
+        trans_btn.click(fn=opus_trans, inputs=[article, lang_select], outputs=result)
 demo.launch()