From fa55fe7670f1d969019244d1d732668b3dac4afb Mon Sep 17 00:00:00 2001
From: Wojciech Jarmosz <wojciech@jarmosz.com>
Date: Sun, 23 May 2021 17:44:24 +0200
Subject: [PATCH] Add model prediction

---
 mlflow_prediction.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlflow_prediction.py b/mlflow_prediction.py
index af7a290..e1a8140 100644
--- a/mlflow_prediction.py
+++ b/mlflow_prediction.py
@@ -15,4 +15,4 @@ with open(f'my_model/{input}', 'r') as file:
     json_data = json.load(file)['inputs']
 
 data_into_dict = {idx:x for idx, x in enumerate(json_data)}
-print(model.predict(pd.DataFrame([data_into_dict])))
\ No newline at end of file
+print(model(pd.DataFrame([data_into_dict])))
\ No newline at end of file