diff --git a/pred_from_mlflow_model.py b/pred_from_mlflow_model.py index c7f91a8..8dc9b79 100644 --- a/pred_from_mlflow_model.py +++ b/pred_from_mlflow_model.py @@ -3,8 +3,8 @@ import json import mlflow as mlf import pandas as pd -model_name = "movies_imdb" -model_version = 1 +model_name = "s430705" +model_version = 8 model = mlf.pyfunc.load_model( model_uri=f"models:/{model_name}/{model_version}"