richylyq commited on
Commit
e3d64dc
1 Parent(s): e3a8bf2

update logging positions and enable OPUS

Browse files
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -172,16 +172,16 @@ def nllb_trans(article, target_language):
172
 
173
 
174
  def mbart_trans(article, target_language):
175
- logger.info(f"Article to translate : {article}")
176
- logger.info(f"Chose which translation model: mbart model")
177
- logger.info(f"Language selected: {target_language}")
178
-
179
  result_lang = detect_lang(article)
180
  if target_language == "English":
181
  target_lang = "en"
182
  elif target_language == "Chinese":
183
  target_lang = "zh"
184
 
 
 
 
 
185
  if result_lang != target_lang:
186
  if target_language == "English":
187
  translated = mbart_m2en_model.translate(article, target_lang="en")
@@ -199,16 +199,16 @@ def mbart_trans(article, target_language):
199
 
200
 
201
  def m2m_trans(article, target_language):
202
- logger.info(f"Article to translate : {article}")
203
- logger.info(f"Chose which translation model: m2m model")
204
- logger.info(f"Language selected: {target_language}")
205
-
206
  result_lang = detect_lang(article)
207
  if target_language == "English":
208
  target_lang = "en"
209
  elif target_language == "Chinese":
210
  target_lang = "zh"
211
 
 
 
 
 
212
  if result_lang != target_lang:
213
  if target_language == "English":
214
  translated = m2m_model.translate(article, target_lang)
@@ -226,11 +226,11 @@ def m2m_trans(article, target_language):
226
 
227
 
228
  def translate(article, toolkit, target_language):
229
- # if toolkit == "OPUS":
230
- # translated = opus_trans(article, target_language)
231
  # if toolkit == "NLLB":
232
  # translated = nllb_trans(article, target_language)
233
- if toolkit == "MBART":
234
  translated = mbart_trans(article, target_language)
235
  elif toolkit == "M2M":
236
  translated = m2m_trans(article, target_language)
 
172
 
173
 
174
  def mbart_trans(article, target_language):
 
 
 
 
175
  result_lang = detect_lang(article)
176
  if target_language == "English":
177
  target_lang = "en"
178
  elif target_language == "Chinese":
179
  target_lang = "zh"
180
 
181
+ logger.info(f"Article to translate : {article}")
182
+ logger.info(f"Chose which translation model: mbart model")
183
+ logger.info(f"Language selected: {target_language}")
184
+
185
  if result_lang != target_lang:
186
  if target_language == "English":
187
  translated = mbart_m2en_model.translate(article, target_lang="en")
 
199
 
200
 
201
  def m2m_trans(article, target_language):
 
 
 
 
202
  result_lang = detect_lang(article)
203
  if target_language == "English":
204
  target_lang = "en"
205
  elif target_language == "Chinese":
206
  target_lang = "zh"
207
 
208
+ logger.info(f"Article to translate : {article}")
209
+ logger.info(f"Chose which translation model: m2m model")
210
+ logger.info(f"Language selected: {target_language}")
211
+
212
  if result_lang != target_lang:
213
  if target_language == "English":
214
  translated = m2m_model.translate(article, target_lang)
 
226
 
227
 
228
  def translate(article, toolkit, target_language):
229
+ if toolkit == "OPUS":
230
+ translated = opus_trans(article, target_language)
231
  # if toolkit == "NLLB":
232
  # translated = nllb_trans(article, target_language)
233
+ elif toolkit == "MBART":
234
  translated = mbart_trans(article, target_language)
235
  elif toolkit == "M2M":
236
  translated = m2m_trans(article, target_language)