Ver código fonte

Removed nudenet

TheHamkerCat 5 anos atrás
pai
commit
cbdba53942
2 arquivos alterados com 0 adições e 30 exclusões
  1. 0 1
      requirements.txt
  2. 0 29
      wbb/modules/nudenet.py

+ 0 - 1
requirements.txt

@@ -1,4 +1,3 @@
-nudenet
 requests
 python-arq
 TgCrypto

+ 0 - 29
wbb/modules/nudenet.py

@@ -1,29 +0,0 @@
-from wbb import app
-from pyrogram import filters
-from wbb.utils.errors import capture_err
-from nudenet import NudeDetector
-import os
-
-detector = NudeDetector()
-
-
-@app.on_message(filters.photo & filters.sticker & filters.document)
-async def nudenet_detector(_, message):
-    if message.document:
-        if message.document.mime_type != "image/png" and \
-                message.document.mime_type != "image/jpg" and \
-                message.document.mime_type != "image/jpeg":
-                    return
-    mention = message.from_user.mention
-    file = await message.download()
-    image = detector.detect(file, mode='fast')
-    if image:
-        try:
-            await message.delete()
-        except Exception:
-            await message.reply_text("It's a NSFW image, If i had enough permissions, I would've deleted it")
-            os.remove(file)
-            return
-    await message.reply_text(f"{mention} Sent a NSFW image, so i deleted it.")
-    os.remove(f)
-