add in target_language assignment

app.py

@@ -178,7 +178,10 @@ def mbart_trans(article, target_language):
 
 def m2m_trans(article, target_language):
     result_lang = detect_lang(article)
-    print(result_lang)
+    if target_language == "English":
+        target_language = "en"
+    elif target_language == "Chinese":
+        target_language = "zh"
     if result_lang != target_language:
         if target_language == "English":
             return m2m_model.translate(article, target_lang="en")