add back mbart model
Browse files
app.py
CHANGED
@@ -31,9 +31,9 @@ from easynmt import EasyNMT
|
|
31 |
# tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
|
32 |
# model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
|
33 |
|
34 |
-
#
|
35 |
-
|
36 |
-
|
37 |
|
38 |
# Initialize m2m_100 models
|
39 |
m2m_model = EasyNMT("m2m_100_1.2B")
|
|
|
31 |
# tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
|
32 |
# model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
|
33 |
|
34 |
+
# Initialize mbart50 models
|
35 |
+
mbart_m2en_model = EasyNMT("mbart50_m2en")
|
36 |
+
mbart_en2m_model = EasyNMT("mbart50_en2m")
|
37 |
|
38 |
# Initialize m2m_100 models
|
39 |
m2m_model = EasyNMT("m2m_100_1.2B")
|