Update index.js

index.js

@@ -25,7 +25,7 @@ app.use(async (ctx) => {
     .ensureAlpha()
     .raw()
     .toBuffer({ resolveWithObject: true })
-    const image_phash = await phash(image_buffer)
+  const image_phash = await phash(image_buffer)
   const colors = await extractColors(
     { data: image_raw, width: info.width, height: info.height },
     {
@@ -52,10 +52,10 @@ app.use(async (ctx) => {
     })
     .gif({ dither: 0 })
     .toBuffer()
-    // const model = await nsfw.load(`https://raw.githubusercontent.com/infinitered/nsfwjs/refs/heads/master/models/mobilenet_v2_mid/model.json`, { type: 'graph' })
+  // const model = await nsfw.load(`https://raw.githubusercontent.com/infinitered/nsfwjs/refs/heads/master/models/mobilenet_v2_mid/model.json`, { type: 'graph' })
   //   // const model = await nsfw.load(`https://raw.githubusercontent.com/infinitered/nsfwjs/refs/heads/master/models/mobilenet_v2/model.json`)
   //   // const model = await nsfw.load(`https://s3.amazonaws.com/ir_public/nsfwjscdn/TFJS_nsfw_mobilenet/tfjs_quant_nsfw_mobilenet/model.json`)
-  //   const model = await nsfw.load(`https://raw.githubusercontent.com/infinitered/nsfwjs/refs/heads/master/models/inception_v3/model.json`, { size: 299 })
+  const model = await nsfw.load(`https://raw.githubusercontent.com/infinitered/nsfwjs/refs/heads/master/models/inception_v3/model.json`, { size: 299 })
   const predictions = await model.classifyGif(image_resize, { fps: 1 })
   const detail = new Map()
   predictions
@@ -67,11 +67,11 @@ app.use(async (ctx) => {
         Math.floor(i.probability * 100),
       )
     })
-    const fail = ['hentai', 'porn', 'sexy']
+  const fail = ['hentai', 'porn', 'sexy']
   const result = [...detail]
     .filter((i) => i.pop() > 7)
     .some((i) => fail.includes(i.pop()))
-    tf.dispose()
+  tf.dispose()
   model.model.dispose()
   ctx.body = {
     nsfw: Object.assign({ result }, Object.fromEntries(detail)),