From 1ea84c709d9e7636abcaa1a53f377ab9e539833f Mon Sep 17 00:00:00 2001 From: Dawid Date: Sun, 23 May 2021 19:10:46 +0200 Subject: [PATCH] change model name directory --- pred_from_mlflow_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pred_from_mlflow_model.py b/pred_from_mlflow_model.py index 8032e03..ccb40cb 100644 --- a/pred_from_mlflow_model.py +++ b/pred_from_mlflow_model.py @@ -8,7 +8,7 @@ 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: +with open('movies_imdb2/input_example.json', 'r') as datafile: data = json.load(datafile)['inputs'] input_dictionary = {idx:x for idx, x in enumerate(data)}