change to standard logging lib

app.py

@@ -24,12 +24,10 @@ from gradio.themes.utils import colors, fonts, sizes
 import argparse
 
 import langid
-from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer, logging
+from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
 from easynmt import EasyNMT
 
-# Initialize logging
-logging.set_verbosity_info()
-logger = logging.get_logger("transformers")
+import logging
 
 # # Initialize nllb-200 models
 # tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
@@ -38,11 +36,11 @@ logger = logging.get_logger("transformers")
 # Initialize mbart50 models
 mbart_m2en_model = EasyNMT("mbart50_m2en")
 mbart_en2m_model = EasyNMT("mbart50_en2m")
-logger.info("mbart50 models initialized")
+logging.info("mbart50 models initialized")
 
 # Initialize m2m_100 models
 m2m_model = EasyNMT("m2m_100_1.2B")
-logger.info("m2m_100 models initialized")
+logging.info("m2m_100 models initialized")
 
 
 class myTheme(Base):
@@ -108,7 +106,7 @@ def detect_lang(article):
     """
 
     result_lang = langid.classify(article)
-    logger.info(f"language detected as {result_lang}")
+    logging.info(f"language detected as {result_lang}")
     return result_lang[0]
 
 
@@ -178,21 +176,21 @@ def mbart_trans(article, target_language):
     elif target_language == "Chinese":
         target_lang = "zh"
 
-    logger.info(f"Article to translate : {article}")
-    logger.info(f"Chose which translation model: mbart model")
-    logger.info(f"Language selected: {target_language}")
+    logging.info(f"Article to translate : {article}")
+    logging.info(f"Chose which translation model: mbart model")
+    logging.info(f"Language selected: {target_language}")
 
     if result_lang != target_lang:
         if target_language == "English":
             translated = mbart_m2en_model.translate(article, target_lang="en")
-            logger.info(f"Translated Result: {translated}")
+            logging.info(f"Translated Result: {translated}")
             return translated
         else:
             translated = mbart_en2m_model.translate(article, target_lang="zh")
-            logger.info(f"Translated Result: {translated}")
+            logging.info(f"Translated Result: {translated}")
             return translated
     else:
-        logger.warning(
+        logging.warning(
             "Error: You chose the same language as the article detected language. Please reselect language and try again."
         )
         return "Error: You chose the same language as the article detected language. Please reselect language and try again."
@@ -205,16 +203,16 @@ def m2m_trans(article, target_language):
     elif target_language == "Chinese":
         target_lang = "zh"
 
-    logger.info(f"Article to translate : {article}")
-    logger.info(f"Chose which translation model: m2m model")
-    logger.info(f"Language selected: {target_language}")
+    logging.info(f"Article to translate : {article}")
+    logging.info(f"Chose which translation model: m2m model")
+    logging.info(f"Language selected: {target_language}")
 
     if result_lang != target_lang:
         translated = m2m_model.translate(article, target_lang)
-        logger.info(f"Translation Result: {translated}")
+        logging.info(f"Translation Result: {translated}")
         return translated
     else:
-        logger.warning(
+        logging.warning(
             f"Error: You chose the same language as the article detected language. Please reselect language and try again."
         )
         return "Error: You chose the same language as the article detected language. Please reselect language and try again."