diff --git a/requirements.txt b/requirements.txt
index 0f8a1c9..4745802 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,7 @@ pandas==0.25.3
 jupyterlab==1.2.21 
 scikit-learn==0.21.3
 matplotlib==3.1.2
-mlflow==1.4.0
+mlflow==2.3.1
 torch==1.3.1+cpu 
 torchvision==0.4.2+cpu 
 xgboost==0.90
\ No newline at end of file