diff --git a/prediction-s444409.py b/prediction-s444409.py
index e72d0eb..093cf89 100644
--- a/prediction-s444409.py
+++ b/prediction-s444409.py
@@ -3,7 +3,7 @@ import mlflow
 import sys
 import numpy as np
 
-#input = sys.argv[1]
+input = sys.argv[1]
 
 logged_model = 'mlruns/1/70439eb482b54d56b54b0ecc6f1ca96f/artifacts/s444409'
 loaded_model = mlflow.pyfunc.load_model(logged_model)