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

minor fixes for language and removing OPUS model

Browse files
Files changed (1) hide show
  1. app.py +11 -7
app.py CHANGED
@@ -166,8 +166,12 @@ def nllb_trans(article, target_language):
166
 
167
  def mbart_trans(article, target_language):
168
  result_lang = detect_lang(article)
 
 
 
 
169
 
170
- if result_lang != target_language:
171
  if target_language == "English":
172
  return mbart_m2en_model.translate(article, target_lang="en")
173
  else:
@@ -184,17 +188,17 @@ def m2m_trans(article, target_language):
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":
189
- return m2m_model.translate(article, target_lang="zh")
190
  else:
191
  return "Error: You chose the same language as the article detected language. Please reselect language and try again."
192
 
193
 
194
  def translate(article, toolkit, target_language):
195
- if toolkit == "OPUS":
196
- translated = opus_trans(article, target_language)
197
- elif toolkit == "NLLB":
198
  translated = nllb_trans(article, target_language)
199
  elif toolkit == "MBART":
200
  translated = mbart_trans(article, target_language)
@@ -209,7 +213,7 @@ myTheme = myTheme()
209
  with gr.Blocks(theme=myTheme) as demo:
210
  article = gr.Textbox(label="Article")
211
  toolkit_select = gr.Radio(
212
- ["OPUS", "NLLB", "MBART", "M2M"], label="Select Translation Model", value="OPUS"
213
  )
214
  lang_select = gr.Radio(["English", "Chinese"], label="Select Desired Language")
215
  result = gr.Textbox(label="Translated Result")
 
166
 
167
  def mbart_trans(article, target_language):
168
  result_lang = detect_lang(article)
169
+ if target_language == "English":
170
+ target_lang = "en"
171
+ elif target_language == "Chinese":
172
+ target_lang = "zh"
173
 
174
+ if result_lang != target_lang:
175
  if target_language == "English":
176
  return mbart_m2en_model.translate(article, target_lang="en")
177
  else:
 
188
  target_lang = "zh"
189
  if result_lang != target_lang:
190
  if target_language == "English":
191
+ return m2m_model.translate(article, target_lang)
192
  elif target_language == "Chinese":
193
+ return m2m_model.translate(article, target_lang)
194
  else:
195
  return "Error: You chose the same language as the article detected language. Please reselect language and try again."
196
 
197
 
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)
 
213
  with gr.Blocks(theme=myTheme) as demo:
214
  article = gr.Textbox(label="Article")
215
  toolkit_select = gr.Radio(
216
+ ["NLLB", "MBART", "M2M"], label="Select Translation Model", value="OPUS"
217
  )
218
  lang_select = gr.Radio(["English", "Chinese"], label="Select Desired Language")
219
  result = gr.Textbox(label="Translated Result")