Rewrite how tensorflow works

This commit is contained in:
Dragon Fire
2024-05-01 01:08:33 -04:00
parent 2bd3f6fcc6
commit 857105ac88
14 changed files with 81 additions and 80 deletions
-17
View File
@@ -2,7 +2,6 @@ const { ActionRowBuilder, ButtonBuilder, PermissionFlagsBits, ButtonStyle, Compo
const crypto = require('crypto');
const request = require('node-superfetch');
const fs = require('fs');
const tf = require('@tensorflow/tfjs-node');
let parseDomain;
let ParseResultType;
import('parse-domain').then(loadedModule => {
@@ -244,22 +243,6 @@ module.exports = class Util {
return str;
}
static async isImageNSFW(model, image, bool = true) {
const img = await tf.node.decodeImage(image, 3);
const predictions = await model.classify(img);
img.dispose();
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 predictions;
}
static async reactIfAble(msg, user, emoji, fallbackEmoji) {
const dm = !msg.guild;
if (!emoji) emoji = fallbackEmoji;