diff --git a/plugins/face/.env b/plugins/face/.env new file mode 100644 index 00000000..09483c0a --- /dev/null +++ b/plugins/face/.env @@ -0,0 +1,2 @@ +TF_FORCE_GPU_ALLOW_GROWTH=true +#CUDA_VISIBLE_DEVICES=0,2 diff --git a/plugins/face/package.json b/plugins/face/package.json index 7f556c3b..448cfd6b 100644 --- a/plugins/face/package.json +++ b/plugins/face/package.json @@ -10,7 +10,7 @@ "socket.io": "^2.0.4", "face-api.js": "^0.22.2", "canvas": "^2.1.0", - "@tensorflow/tfjs-node-gpu": "^0.1.21" + "dotenv": "^8.2.0" }, "devDependencies": {}, "scripts": { diff --git a/plugins/face/shinobi-face.js b/plugins/face/shinobi-face.js index c10108ad..b384389f 100644 --- a/plugins/face/shinobi-face.js +++ b/plugins/face/shinobi-face.js @@ -10,6 +10,7 @@ // Base Init >> var fs = require('fs'); var config = require('./conf.json') +var dotenv = require('dotenv').config() var s try{ s = require('../pluginBase.js')(__dirname,config)