Update index.js

index.js

@@ -337,9 +337,21 @@ async function waifu2x(image, formats) {
 }
 async function check_nsfw(buffer) {
   const model = await nsfwjs.load() // To load a local model, nsfw.load('file://./path/to/model/')
-  // Image must be in tf.tensor3d format
-  // you can convert image to tf.tensor3d with tf.node.decodeImage(Uint8Array,channels)
-  const image = await tfjs.node.decodeImage(buffer,3)
+  const convert = async (img) => {
+  // Decoded image in UInt8 Byte array
+    const image = await jpeg.decode(img, { useTArray: true })
+
+    const numChannels = 3
+    const numPixels = image.width * image.height
+    const values = new Int32Array(numPixels * numChannels)
+
+    for (let i = 0; i < numPixels; i++)
+      for (let c = 0; c < numChannels; ++c)
+        values[i * numChannels + c] = image.data[i * 4 + c]
+
+    return tf.tensor3d(values, [image.height, image.width, numChannels], 'int32')
+  },3)
+  const image = await convert(buffer)
   const predictions = await model.classify(image)
   image.dispose();
   const results = predictions.map(v => {