diff --git a/requirements.txt b/requirements.txt
index b1b816d..29999a9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,4 +10,5 @@ scipy
 torch>=1.9
 torchvision
 timm
-einops
\ No newline at end of file
+einops
+matplotlib