From 7a64af156b662f677844a489ec2201cb5209a375 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Tue, 29 Mar 2022 22:59:01 +0200 Subject: [PATCH] refactoring --- image_prediction/pipeline.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/image_prediction/pipeline.py b/image_prediction/pipeline.py index d680e96..ff9677e 100644 --- a/image_prediction/pipeline.py +++ b/image_prediction/pipeline.py @@ -2,21 +2,20 @@ import os from funcy import rcompose -from image_prediction.formatter.formatters.info_formatter import EnumFormatter - -os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3" - from image_prediction.classifier.classifier import Classifier from image_prediction.classifier.image_classifier import ImageClassifier from image_prediction.config import CONFIG from image_prediction.estimator.adapter.adapter import EstimatorAdapter from image_prediction.extractor_classifier.extractor_classifier import ExtractorClassifier +from image_prediction.formatter.formatters.info_formatter import EnumFormatter from image_prediction.image_extractor.extractors.parsable import ParsablePDFImageExtractor from image_prediction.locations import MLRUNS_DIR from image_prediction.model_loader.loader import ModelLoader from image_prediction.model_loader.loaders.mlflow import MlflowConnector from image_prediction.redai_adapter.mlflow import MlflowModelReader +os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3" + def get_image_classifier(): model_loader = ModelLoader(MlflowConnector(MlflowModelReader(MLRUNS_DIR)))