From 0dd47886dd74a9cf12e2b34e3c1a1728e5889167 Mon Sep 17 00:00:00 2001 From: Kamila Date: Sun, 15 May 2022 12:56:40 +0200 Subject: [PATCH] mlflow attempt task 2 prediction --- predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/predict.py b/predict.py index af22b09..9d7da8c 100644 --- a/predict.py +++ b/predict.py @@ -9,7 +9,7 @@ file.extractall('./ml') input = str((sys.argv[1:])[0]) PATH = "ml/mlruns/1/f65f936936024133a2c03e1e486ba9cf/artifacts/model/" -model =mlflow.pyfunc.load_model(f"{PATH}/MLmodel") +model =mlflow.pyfunc.load_model(f"{PATH}") with open(f'[PATH]/{input}', 'r') as file: json_data = json.load(file)