diff --git a/pyproject.toml b/pyproject.toml
index 9f5dea3..4a0d3dd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -14,6 +14,7 @@ requires-python = ">= 3.9"
 dependencies = [
     "flatbuffers>=2.0",
     "numpy>=1.23.5,<2",
+    "onnxruntime", 
     "onnxruntime-gpu>=1.18.0",
     "Pillow>=10.1.0",
     "protobuf>=3.20.3",
@@ -21,7 +22,7 @@ dependencies = [
     "rich>=13.7.1",
     "loguru>=0.7.2",
     "adb-shell[async]==0.4.4",
-    "PyYAML",  
+    "PyYAML",
 ]
 
 [project.optional-dependencies]