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

Fix

Browse Source
This commit is contained in:
Dragon Fire
2021-03-22 18:13:06 -04:00
parent a8657441d5
commit c7f43ec74d
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
@@ -226,7 +226,7 @@ module.exports = class Util {
height: dimensions.height height: dimensions.height
}; };
const img = await tf.node.decodeImage(data, 3); const img = await tf.node.decodeImage(data, 3);
const predictions = await model.classify(image, 1); const predictions = await model.classify(img, 1);
img.dispose(); img.dispose();
return bool ? predictions[0] !== 'Neutral' && predictions[0] !== 'Drawing' : predictions[0]; return bool ? predictions[0] !== 'Neutral' && predictions[0] !== 'Drawing' : predictions[0];
} }