diff --git a/predict.py b/predict.py
index 760b4ca..1ba4624 100644
--- a/predict.py
+++ b/predict.py
@@ -1,6 +1,6 @@
 import json
 import mlflow
-import pandas as pd
+import sys
 
 input = sys.argv[1]