Update app.py

app.py

@@ -26,10 +26,12 @@ choices = os.popen('tesseract --list-langs').read().split('\n')[1:-1]
 
 
 # 翻译模型选择
-def model_choice(src='trans_src', trg='trans_trg'):
+def model_choice(src, trg):
     # 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}"  # 模型名称
+    # https://huggingface.co/unicamp-dl/translation-pt-en-t5
+    # https://huggingface.co/unicamp-dl/translation-en-pt-t5
+    model_name = f"unicamp-dl/translation-{src}-{trg}-t5"  # 模型名称
 
     tokenizer = MarianTokenizer.from_pretrained(model_name)  # 分词器
     model = MarianMTModel.from_pretrained(model_name)  # 模型
@@ -177,6 +179,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
-
-    #unicamp-dl/translation-pt-en-t5
+    main()