diff --git a/setup.py b/setup.py
index b723f6b..b39ae5d 100644
--- a/setup.py
+++ b/setup.py
@@ -22,6 +22,14 @@
     ],
     package_dir={"": "src"},
     packages=setuptools.find_packages(where="src"),
-    python_requires=">=3.6",
+    install_requires=[
+          'pyyaml',
+          'regex',
+          'matplotlib',
+          'plotly',
+          'scipy',
+          'scikit-learn',
+      ],
+    python_requires=">=3.6, <3.12",
     package_data={'prompt_engine': ['utils/encoder.json', 'utils/vocab.bpe']},
 )
\ No newline at end of file