From a1b13aa789cbef5615a3003a449013249deb4459 Mon Sep 17 00:00:00 2001 From: Maciej Czajka Date: Wed, 11 May 2022 20:17:19 +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 0fa1515..cf91122 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'][0]], dtype=np.float64).reshape( + input_example = np.array([data['inputs'][0]], dtype=np.float64) loaded_model.predict(input_example) \ No newline at end of file