Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,8 @@ choices = os.popen('tesseract --list-langs').read().split('\n')[1:-1]
|
|
18 |
|
19 |
# Função de escolha do modelo de tradução
|
20 |
def model_choice(src, trg):
|
21 |
-
|
|
|
22 |
tokenizer = MarianTokenizer.from_pretrained(model_name)
|
23 |
model = MarianMTModel.from_pretrained(model_name)
|
24 |
return tokenizer, model
|
|
|
18 |
|
19 |
# Função de escolha do modelo de tradução
|
20 |
def model_choice(src, trg):
|
21 |
+
#unicamp-dl/translation-pt-en-t5
|
22 |
+
model_name = f"unicamp-dl/translation-{src}-{trg}-t5"
|
23 |
tokenizer = MarianTokenizer.from_pretrained(model_name)
|
24 |
model = MarianMTModel.from_pretrained(model_name)
|
25 |
return tokenizer, model
|