Update app.py
Browse files
app.py
CHANGED
|
@@ -4,8 +4,6 @@ import torch
|
|
| 4 |
|
| 5 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
| 6 |
|
| 7 |
-
tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
|
| 8 |
-
|
| 9 |
LANG_CODES = {
|
| 10 |
"English":"en",
|
| 11 |
"Toki Pona":"tl",
|
|
@@ -22,16 +20,22 @@ def translate(text, src_lang, tgt_lang, candidates:int):
|
|
| 22 |
|
| 23 |
if tgt == tl and src == en:
|
| 24 |
model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
|
|
|
|
| 25 |
elif tgt == en and src == tl:
|
| 26 |
model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
|
|
|
|
| 27 |
elif tgt == en and src == en:
|
| 28 |
model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
|
|
|
|
| 29 |
elif tgt == tl and src == tl:
|
| 30 |
model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
|
|
|
|
| 31 |
elif tgt == en and src == ro:
|
| 32 |
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M").to(device)
|
|
|
|
| 33 |
elif tgt == ro and src == en:
|
| 34 |
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M").to(device)
|
|
|
|
| 35 |
|
| 36 |
tokenizer.src_lang = src
|
| 37 |
tokenizer.tgt_lang = tgt
|
|
|
|
| 4 |
|
| 5 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
| 6 |
|
|
|
|
|
|
|
| 7 |
LANG_CODES = {
|
| 8 |
"English":"en",
|
| 9 |
"Toki Pona":"tl",
|
|
|
|
| 20 |
|
| 21 |
if tgt == tl and src == en:
|
| 22 |
model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
|
| 23 |
+
tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
|
| 24 |
elif tgt == en and src == tl:
|
| 25 |
model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
|
| 26 |
+
tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
|
| 27 |
elif tgt == en and src == en:
|
| 28 |
model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
|
| 29 |
+
tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
|
| 30 |
elif tgt == tl and src == tl:
|
| 31 |
model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
|
| 32 |
+
tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
|
| 33 |
elif tgt == en and src == ro:
|
| 34 |
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M").to(device)
|
| 35 |
+
tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
|
| 36 |
elif tgt == ro and src == en:
|
| 37 |
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M").to(device)
|
| 38 |
+
tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
|
| 39 |
|
| 40 |
tokenizer.src_lang = src
|
| 41 |
tokenizer.tgt_lang = tgt
|