minor update to app.py

app.py

@@ -53,14 +53,16 @@ def opus_trans(article, target_language):
     Returns:
         string: translated piece of article based off target_language
     """
-    if target_language == "en":
+    if target_language == "English":
         result_lang = "zh"
-    elif target_language == "zh":
+        target_lang = "en"
+    elif target_language == "Chinese":
         result_lang = "en"
+        target_lang = "zh"
     # result_lang = detect_lang(article, target_language)
 
-    task_name = f"translation_{result_lang}_to_{target_language}"
-    model_name = f"Helsinki-NLP/opus-mt-{result_lang}-{target_language}"
+    task_name = f"translation_{result_lang}_to_{target_lang}"
+    model_name = f"Helsinki-NLP/opus-mt-{result_lang}-{target_lang}"
     translator = pipeline(task_name, model=model_name, tokenizer=model_name)
     translated = translator(article)[0]["translation_text"]
     print(translated)