Spaces:
Paused
Paused
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,
|