Update index.js

index.js

@@ -167,10 +167,16 @@ app.post('/api/nsfw-check', async (req, res) => {
             responseType: "arraybuffer"
           })
           const response = await check_nsfw(data_img.data)
-          res.json(response)
+          res.json({
+            status: "ok",
+            result: response
+          })
         } else if (images && typeof images == 'string' && isBase64(images)) {
 	   	  const response = await check_nsfw(Buffer.from(images, "base64"))
-          res.json(response)
+          res.json({
+            status: "ok",
+            result: response
+          })
         } else {
           res.json({
             success: false, message: 'No url or base64 detected!!' 
@@ -336,9 +342,12 @@ async function check_nsfw(buffer) {
   const image = await tfjs.node.decodeImage(buffer,3)
   const predictions = await model.classify(image)
   image.dispose() // Tensor memory must be managed explicitly (it is not sufficient to let a tf.Tensor go out of scope for its memory to be released).
-  return prediction.map(v => {
-    class_name: v.className,
-    probability: v.probability,
-    probability_percent: (v.probability * 100).toFixed(2)
+  cosnt results = prediction.map(v => {
+    return {
+      class_name: v.className,
+      probability: v.probability,
+      probability_percent: (v.probability * 100).toFixed(2)
+    }
   })
+  return results
 }