The accuracy is much less than the site about nsfwjs HOT 1 OPEN

public/
  model/
    model.json
    group1-shard1of1

import * as nsfwjs from 'nsfwjs'
import * as tf from '@tensorflow/tfjs'
import * as jpeg from 'jpeg-js'
const sharp = require('sharp')

//I used @UploadedFile from '@nestjs/common' to handle image uploads.
interface MulterFile {
  fieldname: string
  originalname: string
  encoding: string
  mimetype: string
  buffer: Buffer
  size: number
}

async isCensorableImage(file: MulterFile) {

        const imageToTensor = async (rawImageData: ArrayBuffer): Promise<tf.Tensor3D | ImageData> => {
            rawImageData = await sharp(rawImageData).raw().jpeg().toBuffer()
            const decoded = jpeg.decode(rawImageData); //This is key for the prediction to work well
            const { width, height, data } = decoded
            const buffer = new Uint8Array(width * height * 3);
            let offset = 0;
            for (let i = 0; i < buffer.length; i += 3) {
                buffer[i] = data[offset];
                buffer[i + 1] = data[offset + 1];
                buffer[i + 2] = data[offset + 2];

                offset += 4;
            }

            return tf.tensor3d(buffer, [height, width, 3]);
        }
        await tf.ready();
        tf.env().set('PROD', true) //Disable logs
        const nsfwModel = await nsfwjs.load('http://localhost:8000/model/', { size: 224 }); // Or nsfwjs.load()
        const imageTensor = await imageToTensor(file.buffer);
        const predictionsArray = await nsfwModel.classify(imageTensor)
        console.log(predictionsArray)
        
        return true; //Your logic here checking the predictionsArray
    }