Update app.py

app.py

@@ -26,10 +26,10 @@ choices = os.popen('tesseract --list-langs').read().split('\n')[1:-1]
 
 
 # 翻译模型选择
-def model_choice(src="en", trg="zh"):
-    # https://huggingface.co/Helsinki-NLP/opus-mt-zh-en
-    # https://huggingface.co/Helsinki-NLP/opus-mt-en-zh
-    model_name = f"Helsinki-NLP/opus-mt-{src}-{trg}"  # 模型名称
+def model_choice(src="en", trg="pt"):
+    # https://huggingface.co/Helsinki-NLP/opus-mt-tc-big-pt-en
+    # https://huggingface.co/Helsinki-NLP/opus-mt-tc-big-en-pt
+    model_name = f"Helsinki-NLP/opus-mt-tc-big-{src}-{trg}"  # 模型名称
 
     tokenizer = MarianTokenizer.from_pretrained(model_name)  # 分词器
     model = MarianMTModel.from_pretrained(model_name)  # 模型
@@ -134,9 +134,9 @@ def main():
                     with gr.Row():
                         outputs_text = gr.Textbox(label="Extract content", lines=20)
                     with gr.Row():
-                        inputs_transStyle = gr.Radio(choices=["zh-en", "en-zh"],
+                        inputs_transStyle = gr.Radio(choices=["pt-en", "en-pt"],
                                                      type="value",
-                                                     value="zh-en",
+                                                     value="pt-en",
                                                      label='translation mode')
                     with gr.Row():
                         clear_text_btn = gr.Button('Clear')