Merge branch 'firefox-main' of https://github.com/alganzory/haramblur …

…into firefox-main
alganzory · Mar 22, 2024 · 3a1b7dc · 3a1b7dc
2 parents 84bc4ab + 4aab008
commit 3a1b7dc
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/offscreen.js b/src/offscreen.js
@@ -97,7 +97,11 @@ const runDetection = async (img, isVideo = false) => {
 		detector.human.tf.dispose(tensor);
 		return "nsfw";
 	}
-	if (!settings.shouldDetectGender()) return false; // no need to run gender detection if it's not enabled
+	if (!settings.shouldDetectGender()) 
+	{
+		detector.human.tf.dispose(tensor);
+		return false; // no need to run gender detection if it's not enabled
+	}
 	const predictions = await detector.humanModelClassify(tensor);
 	// console.log("offscreen human result", predictions);
 	detector.human.tf.dispose(tensor);