richylyq commited on
Commit
72801d9
1 Parent(s): c603514

enable mbart

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- translated = nllb_trans(article, target_language)
203
- elif toolkit == "MBART":
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)