enable mbart
Browse files
app.py
CHANGED
@@ -198,9 +198,9 @@ def m2m_trans(article, target_language):
|
|
198 |
def translate(article, toolkit, target_language):
|
199 |
# if toolkit == "OPUS":
|
200 |
# translated = opus_trans(article, target_language)
|
201 |
-
if toolkit == "NLLB":
|
202 |
-
|
203 |
-
|
204 |
translated = mbart_trans(article, target_language)
|
205 |
elif toolkit == "M2M":
|
206 |
translated = m2m_trans(article, target_language)
|
|
|
198 |
def translate(article, toolkit, target_language):
|
199 |
# if toolkit == "OPUS":
|
200 |
# translated = opus_trans(article, target_language)
|
201 |
+
# if toolkit == "NLLB":
|
202 |
+
# translated = nllb_trans(article, target_language)
|
203 |
+
if toolkit == "MBART":
|
204 |
translated = mbart_trans(article, target_language)
|
205 |
elif toolkit == "M2M":
|
206 |
translated = m2m_trans(article, target_language)
|