diff --git a/pyproject.toml b/pyproject.toml
index 497b1a7f..5dec5964 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -60,6 +60,7 @@ sentence-transformers = "*"
 peft = "*"
 psutil = "*"
 timm = "*"
+supervision = "*"
 
 [tool.poetry.dev-dependencies]
 black = "23.3.0"
diff --git a/requirements.txt b/requirements.txt
index c44dc8fc..82236888 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -32,4 +32,5 @@ mkdocs-glightbox
 pre-commit==3.6.2
 psutil
 black
-tenacity
\ No newline at end of file
+tenacity
+supervision
\ No newline at end of file