diff --git a/predict.py b/predict.py
index 07ed448..9734269 100644
--- a/predict.py
+++ b/predict.py
@@ -1,6 +1,7 @@
 import json
 import mlflow
 import sys
+Import numpy as np
 
 input = sys.argv[1]