Create lib/isporn.js

lib/isporn.js

@@ -0,0 +1,17 @@
+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;