richylyq commited on
Commit
b2974e9
·
1 Parent(s): e3d64dc

testing opus add back

Browse files
Files changed (1) hide show
  1. app.py +4 -9
app.py CHANGED
@@ -210,14 +210,9 @@ def m2m_trans(article, target_language):
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)
215
- logger.info(f"Translation Result: {translated}")
216
- return translated
217
- elif target_language == "Chinese":
218
- translated = m2m_model.translate(article, target_lang)
219
- logger.info(f"Translation Result: {translated}")
220
- return translated
221
  else:
222
  logger.warning(
223
  f"Error: You chose the same language as the article detected language. Please reselect language and try again."
@@ -243,7 +238,7 @@ myTheme = myTheme()
243
  with gr.Blocks(theme=myTheme) as demo:
244
  article = gr.Textbox(label="Article")
245
  toolkit_select = gr.Radio(
246
- ["MBART", "M2M"], label="Select Translation Model", value="MBART"
247
  )
248
  lang_select = gr.Radio(["English", "Chinese"], label="Select Desired Language")
249
  result = gr.Textbox(label="Translated Result")
 
210
  logger.info(f"Language selected: {target_language}")
211
 
212
  if result_lang != target_lang:
213
+ translated = m2m_model.translate(article, target_lang)
214
+ logger.info(f"Translation Result: {translated}")
215
+ return translated
 
 
 
 
 
216
  else:
217
  logger.warning(
218
  f"Error: You chose the same language as the article detected language. Please reselect language and try again."
 
238
  with gr.Blocks(theme=myTheme) as demo:
239
  article = gr.Textbox(label="Article")
240
  toolkit_select = gr.Radio(
241
+ ["OPUS", "MBART", "M2M"], label="Select Translation Model", value="MBART"
242
  )
243
  lang_select = gr.Radio(["English", "Chinese"], label="Select Desired Language")
244
  result = gr.Textbox(label="Translated Result")