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

NSFW more safe names

Browse Source
This commit is contained in:
Dragon Fire
2021-03-28 22:20:00 -04:00
parent 1701c431a1
commit a4d3e8c142
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
commands/analyze/nsfw.js
+11 -1
View File
@@ -2,6 +2,13 @@ const Command = require('../../structures/Command');
const request = require('node-superfetch');
const { stripIndents } = require('common-tags');
const { isImageNSFW } = require('../../util/Util');
const displayNames = {
Drawing: 'SFW (Drawing)',
Neutral: 'SFW',
Porn: 'NSFW',
Hentai: 'NSFW (Drawing)',
Sexy: 'Suggestive'
};
module.exports = class NsfwCommand extends Command {
constructor(client) {
@@ -29,7 +36,10 @@ module.exports = class NsfwCommand extends Command {
try {
const { body } = await request.get(image);
const predictions = await isImageNSFW(this.client.nsfwModel, body, false);
const formatted = predictions.map(result => `${Math.round(result.probability * 100)}% ${result.className}`);
const formatted = predictions.map(result => {
const percentage = Math.round(result.probability * 100);
return `${percentage}% ${displayNames[result.className]}`
});
return msg.reply(stripIndents`
**This image gives the following results:**
${formatted.join('\n')}