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