Spaces:

richylyq
/

langtrans

Sleeping

richylyq commited on Nov 8, 2023

Commit

b7e6af4

•

1 Parent(s): 3619186

add target lang for mbart english trans

Files changed (1) hide show

app.py CHANGED Viewed

@@ -166,7 +166,7 @@ def mbart_trans(article, target_language):
     if result_lang != target_language:
         if target_language == "English":
-            return mbart_m2en_model.translate(article, target_language="en")
         else:
             return mbart_en2m_model.translate(article, target_lang="zh")

     if result_lang != target_language:
         if target_language == "English":
+            return mbart_m2en_model.translate(article, target_lang="en")
         else:
             return mbart_en2m_model.translate(article, target_lang="zh")