Khrisna commited on
Commit
7e84ef0
·
1 Parent(s): a8b15f5

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +1 -1
index.js CHANGED
@@ -342,7 +342,7 @@ async function check_nsfw(buffer) {
342
  const image = await tfjs.node.decodeImage(buffer,3)
343
  const predictions = await model.classify(image)
344
  image.dispose();
345
- const results = prediction.map(v => {
346
  return {
347
  class_name: v.className,
348
  probability: v.probability,
 
342
  const image = await tfjs.node.decodeImage(buffer,3)
343
  const predictions = await model.classify(image)
344
  image.dispose();
345
+ const results = predictions.map(v => {
346
  return {
347
  class_name: v.className,
348
  probability: v.probability,