richylyq commited on
Commit
9c13f88
1 Parent(s): f64ffeb

change m2m target language setup

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -179,10 +179,10 @@ def mbart_trans(article, target_language):
179
  def m2m_trans(article, target_language):
180
  result_lang = detect_lang(article)
181
  if target_language == "English":
182
- target_language = "en"
183
  elif target_language == "Chinese":
184
- target_language = "zh"
185
- if result_lang != target_language:
186
  if target_language == "English":
187
  return m2m_model.translate(article, target_lang="en")
188
  elif target_language == "Chinese":
 
179
  def m2m_trans(article, target_language):
180
  result_lang = detect_lang(article)
181
  if target_language == "English":
182
+ target_lang = "en"
183
  elif target_language == "Chinese":
184
+ target_lang = "zh"
185
+ if result_lang != target_lang:
186
  if target_language == "English":
187
  return m2m_model.translate(article, target_lang="en")
188
  elif target_language == "Chinese":