diff --git a/requirements.txt b/requirements.txt
index ab39f1b..4d832e4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,5 +10,5 @@ tensorflow-datasets==4.1.0
 tensorflow-estimator==2.5.0
 tensorflow-metadata==0.25.0
 Pillow==8.3.2
-mlflow==1.12.1
+mlflow==2.3.1
 greenlet==0.4.16