From edba1a08b97cb9a87861dc75c32f0a9caeb73145 Mon Sep 17 00:00:00 2001 From: Maciej Czajka Date: Wed, 11 May 2022 20:09:13 +0200 Subject: [PATCH] update predict.py --- predict.py | 1 + 1 file changed, 1 insertion(+) diff --git a/predict.py b/predict.py index 3e03813..07ed448 100644 --- a/predict.py +++ b/predict.py @@ -10,5 +10,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) loaded_model.predict(data['inputs']) \ No newline at end of file