test m2m
Browse files
app.py
CHANGED
|
@@ -27,13 +27,13 @@ import langid
|
|
| 27 |
from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
|
| 28 |
from easynmt import EasyNMT
|
| 29 |
|
| 30 |
-
# Initialize nllb-200 models
|
| 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")
|
|
|
|
| 27 |
from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
|
| 28 |
from easynmt import EasyNMT
|
| 29 |
|
| 30 |
+
# # Initialize nllb-200 models
|
| 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")
|