From d76fefd3ff0c4425defca4db218ce4a84c6053f3 Mon Sep 17 00:00:00 2001
From: Matthias Bisping <matthias.bisping@iqser.com>
Date: Wed, 16 Mar 2022 14:00:39 +0100
Subject: [PATCH] logger refactoring

---
 config.yaml               | 1 -
 image_prediction/flask.py | 1 +
 src/serve.py              | 6 +++---
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/config.yaml b/config.yaml
index 77ef6f1..dbc5b87 100644
--- a/config.yaml
+++ b/config.yaml
@@ -26,4 +26,3 @@ filters:
     max: $MAX_IMAGE_FORMAT|10  # Maximum permissible
 
   min_confidence: $MIN_CONFIDENCE|0.5  # Minimum permissible prediction confidence
-
diff --git a/image_prediction/flask.py b/image_prediction/flask.py
index c5e8e3d..34f8a29 100644
--- a/image_prediction/flask.py
+++ b/image_prediction/flask.py
@@ -6,6 +6,7 @@ from image_prediction.utils import get_logger
 
 logger = get_logger()
 
+
 def make_prediction_server(predict_fn: Callable):
 
     app = Flask(__name__)
diff --git a/src/serve.py b/src/serve.py
index ba6cb91..39afd1c 100644
--- a/src/serve.py
+++ b/src/serve.py
@@ -6,15 +6,15 @@ from image_prediction.config import CONFIG
 from image_prediction.flask import make_prediction_server
 from image_prediction.predictor import Predictor
 from image_prediction.response import build_response
+from image_prediction.utils import get_logger
 
-logger = logging.getLogger(__name__)
-logger.setLevel(CONFIG.service.logging_level)
+logger = get_logger()
 
 
 def main():
 
     def predict(pdf):
-        predictions, metadata = predictor.predict_pdf(pdf)
+        predictions, metadata = predictor.predict_pdf(pdf, verbose=CONFIG.service.progressbar)
         response = build_response(predictions, metadata)
         return response