diff --git a/libs/computer/pyproject.toml b/libs/computer/pyproject.toml
index ed438947..22465082 100644
--- a/libs/computer/pyproject.toml
+++ b/libs/computer/pyproject.toml
@@ -25,6 +25,7 @@ requires-python = ">=3.10"
 ui = [
     "gradio>=5.23.3,<6.0.0",
     "python-dotenv>=1.0.1,<2.0.0",
+    "datasets>=3.6.0,<4.0.0",
 ]
 
 [tool.pdm]