diff --git a/prediction-s444409.py b/prediction-s444409.py
index 074a48c..e72d0eb 100644
--- a/prediction-s444409.py
+++ b/prediction-s444409.py
@@ -5,8 +5,8 @@ import numpy as np
 
 #input = sys.argv[1]
 
-# logged_model = 'mlruns/1/70439eb482b54d56b54b0ecc6f1ca96f/artifacts/s444409'
-# loaded_model = mlflow.pyfunc.load_model(logged_model)
+logged_model = 'mlruns/1/70439eb482b54d56b54b0ecc6f1ca96f/artifacts/s444409'
+loaded_model = mlflow.pyfunc.load_model(logged_model)
 
 
 with open('input_example.json') as f: