diff --git a/predict.py b/predict.py
index d4abc41..7715c5b 100644
--- a/predict.py
+++ b/predict.py
@@ -13,4 +13,4 @@ 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
+loaded_model.predict(input_example)
\ No newline at end of file