logging change

image_prediction/utils/logger.py

@@ -12,8 +12,8 @@ def make_logger_getter():
     handler = logging.StreamHandler()
     handler.setLevel(CONFIG.service.logging_level)
 
-    log_format = "[%(levelname)s]: %(message)s"
-    formatter = logging.Formatter(log_format)
+    log_format = "%(asctime)s %(levelname)-8s %(message)s"
+    formatter = logging.Formatter(log_format, datefmt="%Y-%m-%d %H:%M:%S")
 
     handler.setFormatter(formatter)
     logger.addHandler(handler)

scripts/run_pipeline.py

@@ -4,8 +4,11 @@ import os
 from glob import glob
 
 from image_prediction.pipeline import load_pipeline
+from image_prediction.utils import get_logger
 from image_prediction.utils.pdf_annotation import annotate_pdf
 
+logger = get_logger()
+
 
 def parse_args():
     parser = argparse.ArgumentParser()
@@ -21,6 +24,7 @@ def parse_args():
 
 def process_pdf(pipeline, pdf_path):
     with open(pdf_path, "rb") as f:
+        logger.info(f"Processing {pdf_path}")
         predictions = list(pipeline(f.read()))
 
     annotate_pdf(