diff --git a/structures/Tensorflow.js b/structures/Tensorflow.js index 36e21a82..1c7acc25 100644 --- a/structures/Tensorflow.js +++ b/structures/Tensorflow.js @@ -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; }