diff --git a/setup.py b/setup.py
index 2d7303f..bc1497a 100644
--- a/setup.py
+++ b/setup.py
@@ -11,6 +11,8 @@
             'pandas',
             'scipy',
             'scikit-learn',
-            'torch'
+            'torch',
+            'matplotlib',
+            'seaborn'
       ],
       zip_safe=False)
\ No newline at end of file