diff --git a/aimodel/pyproject.toml b/aimodel/pyproject.toml
index 5586745..452fa0c 100644
--- a/aimodel/pyproject.toml
+++ b/aimodel/pyproject.toml
@@ -31,18 +31,7 @@ maintainers = [
 
 requires-python = ">= 3.10"
 
-dependencies = [
-	"tensorflow>=2.4",
-	# "tensorflow_addons",
-	"keras_cv",
-	"silence_tensorflow",
-	"loguru",
-	"umap-learn[plot]",
-	"seaborn",
-	"matplotlib",
-	"numpy",
-	"pandas"
-]
+dependencies = { file = ["requirements.txt"] }
 
 [project.urls]
 Homepage = "https://github.com/sbrl/research-rainfallradar"
diff --git a/aimodel/requirements.txt b/aimodel/requirements.txt
index 195b75b..0a7ba0b 100644
--- a/aimodel/requirements.txt
+++ b/aimodel/requirements.txt
@@ -1,5 +1,7 @@
+setuptools # Dependency of tensorflow that isn't mentioned >_<
 tensorflow>=2.4
-tensorflow_addons
+# tensorflow_addons
+keras_cv
 silence_tensorflow
 loguru
 umap-learn[plot]