diff --git a/package.json b/package.json
index 25258753..401e3334 100644
--- a/package.json
+++ b/package.json
@@ -100,11 +100,10 @@
   "dependencies": {
     "@aws-sdk/client-s3": "^3.427.0",
     "@tensorflow-models/universal-sentence-encoder": "^1.3.3",
-    "@tensorflow/tfjs": "^4.22.0",
     "@tensorflow/tfjs-backend-cpu": "^4.22.0",
+    "@tensorflow/tfjs-backend-webgl": "^4.22.0",
     "@tensorflow/tfjs-converter": "^4.22.0",
     "@tensorflow/tfjs-core": "^4.22.0",
-    "@tensorflow/tfjs-layers": "^4.22.0",
     "buffer": "^6.0.3",
     "commander": "^14.0.0",
     "omelette": "^0.4.17",