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

Fix

Browse Source
This commit is contained in:
Dragon Fire
2021-03-22 18:29:34 -04:00
parent 1ab7bd4563
commit bbe2702397
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
commands/analyze/nsfw.js
+2 -1
View File
@@ -28,7 +28,8 @@ module.exports = class NsfwCommand extends Command {
try {
const { body } = await request.get(image);
const prediction = await isImageNSFW(this.client.nsfwModel, body, false);
return msg.reply(`I'm **${prediction.probability}%** sure this image is: **${prediction.className}**.`);
const prob = Math.round(prediction.probability * 100);
return msg.reply(`I'm **${prob}%** sure this image is: **${prediction.className}**.`);
} catch (err) {
return msg.reply(`Oh no, an error occurred: \`${err.message}\`. Try again later!`);
}