Update app.py

app.py

@@ -2,22 +2,35 @@ import gradio as gr
 
 from transformers import pipeline
 
-pipe = pipeline("translation_en_to_zh", model='facebook/m2m100_418M')
 
+model_list = [
+    "facebook/m2m100_418M", 
+    "Helsinki-NLP/opus-mt-en-zh", 
+    "liam168/trans-opus-mt-en-zh",
+    "facebook/nllb-200-3.3B"
+    ]
 
-def translate(text):
+#init
+for model_name in model_list:
+    pipeline("translation_en_to_zh", model=model_name)
+
+def translate(text, model_name):
+    pipe = pipeline("translation_en_to_zh", model=model_name)
     return pipe(text)[0]["translation_text"]
 
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
-            english = gr.Textbox(label="English text")
+            englishtxt_model_input = [
+                gr.Textbox(label="English text"), 
+                gr.Dropdown(model_list, value="facebook/m2m100_418M", label="Model")
+            ]
             translate_btn = gr.Button(value="Translate")
         with gr.Column():
-            chinese = gr.Textbox(label="Chinese Text")
+            chinese_output = gr.Textbox(label="Chinese Text")
 
-    translate_btn.click(translate, inputs=english, outputs=chinese, api_name="translate-to-chinese")
+    translate_btn.click(translate, inputs=englishtxt_model_input, outputs=chinese_output, api_name="translate-to-chinese")
     examples = gr.Examples(examples=["I went to the supermarket yesterday.", "Helen is a good swimmer."],
                            inputs=[english])
 
-demo.launch()
+demo.launch()