richylyq commited on
Commit
9ee6383
1 Parent(s): 87e6706

simplified trans

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -53,6 +53,11 @@ def opus_trans(message, target_lang):
53
  Returns:
54
  string: translated piece of article based off target_lang
55
  """
 
 
 
 
 
56
 
57
  task_name = f"translation_{result_lang}_to_{target_lang}"
58
  model_name = f"Helsinki-NLP/opus-mt-{result_lang}-{target_lang}"
@@ -64,7 +69,6 @@ def opus_trans(message, target_lang):
64
 
65
  article = gr.Textbox()
66
  lang_select = gr.Dropdown(["en", "zh"])
67
- # lang_detect = detect_lang(article["text"], lang_select)
68
  translate = gr.Interface(
69
  opus_trans,
70
  [
 
53
  Returns:
54
  string: translated piece of article based off target_lang
55
  """
56
+ if target_lang == "en":
57
+ result_lang = "zh"
58
+ elif target_lang == "zh":
59
+ result_lang = "en"
60
+ # result_lang = detect_lang(article, target_lang)
61
 
62
  task_name = f"translation_{result_lang}_to_{target_lang}"
63
  model_name = f"Helsinki-NLP/opus-mt-{result_lang}-{target_lang}"
 
69
 
70
  article = gr.Textbox()
71
  lang_select = gr.Dropdown(["en", "zh"])
 
72
  translate = gr.Interface(
73
  opus_trans,
74
  [