From 1a0d9a1c85cc14b889f7320ed82f694d495b0514 Mon Sep 17 00:00:00 2001 From: Dawid Date: Sun, 23 May 2021 19:07:22 +0200 Subject: [PATCH] add default uri --- pred_from_mlflow_model.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/pred_from_mlflow_model.py b/pred_from_mlflow_model.py index 8dc9b79..8032e03 100644 --- a/pred_from_mlflow_model.py +++ b/pred_from_mlflow_model.py @@ -1,15 +1,13 @@ -import mlflow as mlf import json import mlflow as mlf import pandas as pd model_name = "s430705" model_version = 8 - +mlf.set_tracking_uri("http://172.17.0.1:5000") model = mlf.pyfunc.load_model( model_uri=f"models:/{model_name}/{model_version}" ) - with open('movies_imdb/input_example.json', 'r') as datafile: data = json.load(datafile)['inputs']