From 14792760567bfa1caaaad08e6cd25e1435b5a164 Mon Sep 17 00:00:00 2001 From: Maciej Czajka Date: Wed, 11 May 2022 20:15:02 +0200 Subject: [PATCH] update predict.py --- predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/predict.py b/predict.py index 7715c5b..0fa1515 100644 --- a/predict.py +++ b/predict.py @@ -11,6 +11,6 @@ loaded_model = mlflow.pyfunc.load_model(logged_model) with open(f'{logged_model}/'+input) as f: data = json.load(f) - input_example = np.array([data['inputs']], dtype=np.float64) + input_example = np.array([data['inputs'][0]], dtype=np.float64).reshape( loaded_model.predict(input_example) \ No newline at end of file