diff --git a/src/serve.py b/src/serve.py index 8652f5b..848175c 100644 --- a/src/serve.py +++ b/src/serve.py @@ -9,8 +9,7 @@ from image_prediction.pipeline import load_pipeline from image_prediction.utils.banner import load_banner from image_prediction.utils.process_wrapping import wrap_in_process -logger.remove() -logger.add(sink=stdout, level=CONFIG.logging.level) +logger.reconfigure(sink=stdout, level=CONFIG.logging.level) # A component of the processing pipeline (probably tensorflow) does not release allocated memory (see RED-4206).