diff --git a/requirements.txt b/requirements.txt
index 9aa772c..e54ae9f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@
 scikit-learn==0.21.3
 jieba==0.42.1
 bert4keras==0.5.9
-mlflow==1.2.0
+mlflow==2.2.1
 xgboost==0.90
 lightgbm==2.3.0
 gensim==3.8.1
diff --git a/setup.py b/setup.py
index 07051d3..2ccf440 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
     long_description = fh.read()
 
 REQUIRED_PACKAGES = [
-    'scikit-learn==0.21.3','jieba==0.42.1',"bert4keras==0.5.9", "mlflow==1.2.0"
+    'scikit-learn==0.21.3','jieba==0.42.1',"bert4keras==0.5.9", "mlflow==2.2.1"
 ]
 
 setuptools.setup(