Update index.js

index.js

@@ -351,7 +351,7 @@ async function check_nsfw(buffer) {
         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();