diff --git a/requirements.txt b/requirements.txt
index ec06324..12de7da 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,4 +6,5 @@ scikit-image
 scipy
 torch
 torchvision
-tqdm
\ No newline at end of file
+tqdm
+pymatting