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

Fix

Browse Source
This commit is contained in:
Dragon Fire
2021-03-22 21:47:09 -04:00
parent 76f4b018e0
commit d90c757757
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
util/Util.js
+9 -8
View File
@@ -221,15 +221,16 @@ module.exports = class Util {
const img = await tf.node.decodeImage(image, 3);
const predictions = await model.classify(img, 2);
img.dispose();
const results = [];
results.push(predictions[0]);
for (const result of predictions) {
if (result.className === predictions[0].className) continue;
if (result.probability >= predictions[0].probability - 0.05) results.push(result);
if (bool) {
const results = [];
results.push(predictions[0]);
for (const result of predictions) {
if (result.className === predictions[0].className) continue;
if (result.probability >= predictions[0].probability - 0.1) results.push(result);
}
return results.some(result => result.className !== 'Drawing' && result.className !== 'Neutral');
}
return bool
? results.some(result => result.className !== 'Drawing' && result.className !== 'Neutral')
: results;
return predictions;
}
static async reactIfAble(msg, user, emoji, fallbackEmoji) {