diff --git a/pyproject.toml b/pyproject.toml
index 996e3d2..6c25b6e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -45,7 +45,8 @@ dependencies = [
     "huggingface-hub",
     "setuptools",
     "spacy",
-    "tensorflow",
+    "tensorflow<=2.16.2",
+    "tf-keras",
     "google-cloud-vision",
     "colorgram.py",
     "webcolors",