update predict.py

2022-05-11 20:18:32 +02:00 · 2022-05-11 20:18:32 +02:00 · 30a09f0f7d
commit 30a09f0f7d
parent a1b13aa789
1 changed files with 1 additions and 1 deletions
--- 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)