diff --git a/src/serve.py b/src/serve.py index 005cbb2..57749b0 100644 --- a/src/serve.py +++ b/src/serve.py @@ -10,7 +10,6 @@ from image_prediction.utils.banner import show_banner def main(): - logger = get_logger() def predict(pdf): # Keras service_estimator.predict stalls when service_estimator was loaded in different process; @@ -26,7 +25,9 @@ def main(): if __name__ == "__main__": - logging.basicConfig(level=CONFIG.service.logging_level) + logging.basicConfig() + logger = get_logger() + logger = logger.setLevel(CONFIG.service.logging_level) show_banner()