diff --git a/image_classifier_project/requirements.txt b/image_classifier_project/requirements.txt
index bb664d31..30bd10ea 100644
--- a/image_classifier_project/requirements.txt
+++ b/image_classifier_project/requirements.txt
@@ -53,7 +53,7 @@ PySocks==1.6.8
 python-dateutil==2.7.3
 pytz==2018.4
 pywinpty==0.5.3
-PyYAML==3.12
+PyYAML==5.4
 pyzmq==17.0.0
 qtconsole==4.3.1
 requests==2.18.4