From 578f3e5155caaaf1f5ad1def3793fe569cc3d5aa Mon Sep 17 00:00:00 2001
From: Dragon Fire <danielbodendahl@gmail.com>
Date: Mon, 13 May 2024 23:30:44 -0400
Subject: [PATCH] set nsfwjs model directly

---
 structures/Tensorflow.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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;
 	}