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