set nsfwjs model directly

This commit is contained in:
Dragon Fire
2024-05-13 23:30:44 -04:00
parent 22ee2a5497
commit 578f3e5155
+1 -1
View File
@@ -16,7 +16,7 @@ module.exports = class Tensorflow {
}
async loadNSFWJS() {
const nsfwjs = await nsfw.load();
const nsfwjs = await nsfw.load('MobileNetV2');
this.nsfwjs = nsfwjs;
return this.nsfwjs;
}