Khrisna commited on
Commit
7f476e3
·
1 Parent(s): a1c6dd2

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +1 -1
index.js CHANGED
@@ -350,7 +350,7 @@ async function check_nsfw(buffer) {
350
  for (let c = 0; c < numChannels; ++c)
351
  values[i * numChannels + c] = image.data[i * 4 + c]
352
 
353
- return tf.tensor3d(values, [image.height, image.width, numChannels], 'int32')
354
  }
355
  const image = await convert(buffer)
356
  const predictions = await model.classify(image)
 
350
  for (let c = 0; c < numChannels; ++c)
351
  values[i * numChannels + c] = image.data[i * 4 + c]
352
 
353
+ return tfjs.tensor3d(values, [image.height, image.width, numChannels], 'int32')
354
  }
355
  const image = await convert(buffer)
356
  const predictions = await model.classify(image)