diff --git a/predict.py b/predict.py
index 3e03813..07ed448 100644
--- a/predict.py
+++ b/predict.py
@@ -10,5 +10,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']], dtype=np.float64)
 
 loaded_model.predict(data['inputs'])
\ No newline at end of file