diff --git a/mlflow_prediction_registry.py b/mlflow_prediction_registry.py index baebc1f..4753dc4 100644 --- a/mlflow_prediction_registry.py +++ b/mlflow_prediction_registry.py @@ -14,7 +14,7 @@ arguments = sys.argv[1:] mlflow.set_tracking_uri("http://172.17.0.1:5000") client = MlflowClient() version = 1 -model_name = "s434704" +model_name = "s426206" input = str(arguments[0])