Spaces:
Running
Running
Update index.js
Browse files
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 =
|
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,
|