diff --git a/scripts/mlflow_train.py b/scripts/mlflow_train.py
index bd0b9e7..ebca449 100644
--- a/scripts/mlflow_train.py
+++ b/scripts/mlflow_train.py
@@ -12,7 +12,8 @@ from torch import nn
 from torch.utils import data as t_u_data
 
 
-mlflow.set_tracking_uri("http://localhost:5000")
+# mlflow.set_tracking_uri("http://localhost:5000")
+mlflow.set_tracking_uri("http://172.17.0.1:5000")
 mlflow.set_experiment("s478841")