From 2c8b593caddd328f61eff4b036cc16456b5f2adc Mon Sep 17 00:00:00 2001
From: Dragon Fire <danielbodendahl@gmail.com>
Date: Mon, 22 Mar 2021 21:47:29 -0400
Subject: [PATCH] Fix

---
 util/Util.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/util/Util.js b/util/Util.js
index 5e0c06f9..e3b254fd 100644
--- a/util/Util.js
+++ b/util/Util.js
@@ -219,7 +219,7 @@ module.exports = class Util {
 
 	static async isImageNSFW(model, image, bool = true) {
 		const img = await tf.node.decodeImage(image, 3);
-		const predictions = await model.classify(img, 2);
+		const predictions = await model.classify(img);
 		img.dispose();
 		if (bool) {
 			const results = [];