Update app.py

app.py

@@ -18,7 +18,8 @@ choices = os.popen('tesseract --list-langs').read().split('\n')[1:-1]
 
 # Função de escolha do modelo de tradução
 def model_choice(src, trg):
-    model_name = f"Helsinki-NLP/opus-mt-tc-big-{src}-{trg}"
+    #unicamp-dl/translation-pt-en-t5
+    model_name = f"unicamp-dl/translation-{src}-{trg}-t5"
     tokenizer = MarianTokenizer.from_pretrained(model_name)
     model = MarianMTModel.from_pretrained(model_name)
     return tokenizer, model