diff --git a/requirements.txt b/requirements.txt
index 8f8df0b..bbcf159 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,5 +3,5 @@ opencv-python>=4.1.2
 torch>=1.7.0
 torchvision>=0.8.1
 seaborn
-
+pyyaml
 towhee
\ No newline at end of file