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)