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

More accurate check

Browse Source
This commit is contained in:
Dragon Fire
2021-03-22 21:43:11 -04:00
parent 832595c7d0
commit 0d8e8fd4d9
2 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
commands/analyze/nsfw.js
+7 -3
View File
@@ -1,5 +1,6 @@
const Command = require('../../structures/Command');
const request = require('node-superfetch');
const { stripIndents } = require('common-tags');
const { isImageNSFW } = require('../../util/Util');
module.exports = class NsfwCommand extends Command {
@@ -27,9 +28,12 @@ module.exports = class NsfwCommand extends Command {
async run(msg, { image }) {
try {
const { body } = await request.get(image);
const prediction = await isImageNSFW(this.client.nsfwModel, body, false);
const prob = Math.round(prediction.probability * 100);
return msg.reply(`I'm **${prob}%** sure this image is: **${prediction.className}**.`);
const predictions = await isImageNSFW(this.client.nsfwModel, body, false);
const formatted = predictions.map(result => `${Math.round(result.probability * 100)}% ${result.className}`);
return msg.reply(stripIndents`
**This image gives the following results:**
${formatted.join('\n')}
`);
} catch (err) {
return msg.reply(`Oh no, an error occurred: \`${err.message}\`. Try again later!`);
}