21 lines
707 B
Python
21 lines
707 B
Python
from image_prediction.config import CONFIG
|
|
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.pipeline import Pipeline
|
|
from image_prediction.redai_adapter.mlflow import MlflowModelReader
|
|
|
|
|
|
def get_mlflow_model_loader(mlruns_dir):
|
|
model_loader = ModelLoader(MlflowConnector(MlflowModelReader(mlruns_dir)))
|
|
return model_loader
|
|
|
|
|
|
def load_pipeline(**kwargs):
|
|
model_loader = get_mlflow_model_loader(MLRUNS_DIR)
|
|
model_identifier = CONFIG.service.run_id
|
|
|
|
pipeline = Pipeline(model_loader, model_identifier, **kwargs)
|
|
|
|
return pipeline
|