simplified trans
Browse files
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 |
[
|