Spaces:
Paused
Paused
Update index.js
Browse files
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
|
| 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)
|