From 484aa77a51ff40f88b7738f95a1f1d91d6cacfc3 Mon Sep 17 00:00:00 2001 From: Kamila Date: Sun, 15 May 2022 12:58:16 +0200 Subject: [PATCH] mlflow attempt task 2 prediction --- predict.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/predict.py b/predict.py index 5fad2de..6929115 100644 --- a/predict.py +++ b/predict.py @@ -14,5 +14,5 @@ model =mlflow.pyfunc.load_model(f"{PATH}") with open(f'{PATH}{input}', 'r') as file: json_data = json.load(file) - print(f"Input: {json_data['inputs'][0]}") - print(f"Prediction: {model.predict(np.array([json_data['inputs'][0]], dtype=np.float32))}") \ No newline at end of file + print(f"Input: {json_data['inputs']}") + print(f"Prediction: {model.predict(np.array([json_data['inputs']], dtype=np.float32))}") \ No newline at end of file