add logging and test

app.py

@@ -24,9 +24,13 @@ from gradio.themes.utils import colors, fonts, sizes
 import argparse
 
 import langid
-from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
+from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer, logging
 from easynmt import EasyNMT
 
+# Initialize logging
+logging.set_verbosity_info()
+logger = logging.get_logger("transformers")
+
 # # Initialize nllb-200 models
 # tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
 # model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
@@ -34,9 +38,11 @@ from easynmt import EasyNMT
 # Initialize mbart50 models
 mbart_m2en_model = EasyNMT("mbart50_m2en")
 mbart_en2m_model = EasyNMT("mbart50_en2m")
+logger.info("mbart50 models initialized")
 
 # Initialize m2m_100 models
 m2m_model = EasyNMT("m2m_100_1.2B")
+logger.info("m2m_100 models initialized")
 
 
 class myTheme(Base):
@@ -102,6 +108,7 @@ def detect_lang(article):
     """
 
     result_lang = langid.classify(article)
+    logger.info(f"language detected as {result_lang}")
     return result_lang[0]