minor fixes for language and removing OPUS model
Browse files
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 !=
|
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
|
188 |
elif target_language == "Chinese":
|
189 |
-
return m2m_model.translate(article, target_lang
|
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 |
-
|
197 |
-
|
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 |
-
["
|
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")
|