diff --git a/train_model.py b/train_model.py
index d3af166..d648d55 100644
--- a/train_model.py
+++ b/train_model.py
@@ -13,7 +13,7 @@ default_epochs = 5
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-#mlflow.set_tracking_uri("http://172.17.0.1:5000")
+mlflow.set_tracking_uri("http://172.17.0.1:5000")
 mlflow.set_experiment("s444409")