const tf = require('@tensorflow/tfjs-node');
const nsfw = require('nsfwjs');

async function fn(buffer) {
 const model = await nsfw.load();
 const image = await tf.node.decodeImage(buffer,3)
 const predictions = await model.classify(image)
 image.dispose();
 return predictions.map(item => {
    return {
      type: item.className,
      probability: (item.probability * 100).toFixed(2)
    }
 });
}

module.exports = fn;