arthur/xiao
1
0
Fork 0
mirror of https://github.com/arthur-pbty/xiao.git synced 2026-06-03 23:36:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Dragon Fire
2021-03-22 21:47:29 -04:00
parent d90c757757
commit 2c8b593cad
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
util/Util.js
+1 -1
View File
@@ -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 = [];