Update app.py

app.py

@@ -7,6 +7,14 @@ import gradio as gr
 from transformers import MBart50TokenizerFast, MBartForConditionalGeneration
 
 
+language_options = {
+    '简体中文': 'zh_CN',
+    '英文': 'en_XX',
+    '繁体中文': 'zh_TW',
+    '泰文': 'th_TH',
+}
+languages = list(language_options.keys())
+
 class MBartTranslator:
     """MBartTranslator class provides a simple interface for translating text using the MBart language model.
 
@@ -127,18 +135,18 @@ translator = MBartTranslator()
 
 
 def translate(src, dst, content):
-    outputText = translator.translate(content, "zh_CN", "en_XX")
-    return outputText
+    output = translator.translate(content, language_options[src], language_options[dst])
+    return output
 
 
 demo = gr.Interface(
     fn=translate,
     inputs=[
         gr.Dropdown(
-            ["简体中文", "繁体中文", "英文", "泰文"], label="源语言", value="简体中文", show_label=True
+            languages, label="源语言", value=languages[0], show_label=True
         ),
         gr.Dropdown(
-            ["简体中文", "繁体中文", "英文", "泰文"], label="目标语言", value="英文", show_label=True
+            languages, label="目标语言", value=languages[1], show_label=True
         ),
         gr.Text(label='内容')
     ],