diff --git a/Jenkinsfile_coop_registry b/Jenkinsfile_coop_registry new file mode 100644 index 0000000..fa132d8 --- /dev/null +++ b/Jenkinsfile_coop_registry @@ -0,0 +1,19 @@ +pipeline { + agent {dockerfile true, args '-v /tmp/mlruns:/tmp/mlruns'} + parameters { + string( + defaultValue: 'input_example.json', + description: 'Input name', + name: 'INPUT_NAME', + trim: false + ) + } + stages { + stage('Load model and run prediction') { + steps { + sh 'python3 mlflow_prediction_registry.py $INPUT_NAME' + } + } + + } +} \ No newline at end of file diff --git a/mlflow_prediction_registry.py b/mlflow_prediction_registry.py new file mode 100644 index 0000000..baebc1f --- /dev/null +++ b/mlflow_prediction_registry.py @@ -0,0 +1,28 @@ + +import json +import mlflow +from mlflow.tracking import MlflowClient +import mlflow.pyfunc +import torch +import numpy as np +import pandas as pd + +import sys + +arguments = sys.argv[1:] + +mlflow.set_tracking_uri("http://172.17.0.1:5000") +client = MlflowClient() +version = 1 +model_name = "s434704" + +input = str(arguments[0]) + +model = mlflow.pyfunc.load_model( + model_uri=f"models:/{model_name}/{model_version}" +) + +with open(f'{model_name}/{input}', 'r') as file: + json_data = json.load(file) + +print(model(torch.tensor(np.array(json_data['inputs'])).float())) \ No newline at end of file