Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-10 14:58:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

linting

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-07-19 11:33:16 -04:00
parent 74d3f0bdd5
commit 3bf31eb4fd
4 changed files with 17 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/nudenet/nudenet.py
+1 -1
View File
@@ -58,7 +58,7 @@ class NudeDetector:
from modules import shared
global session # pylint: disable=global-statement
self.model_path = hf.hf_hub_download(
self.model_path = model or hf.hf_hub_download(
repo_id='vladmandic/nudenet',
filename='nudenet.onnx',
cache_dir=shared.opts.diffusers_dir,
scripts/nudenet_ext.py
+6 -9
View File
@@ -1,12 +1,9 @@
# built-in imports and third party imports
import gradio as gr
# import required modules from sdnext
from modules import scripts, scripts_postprocessing, script_callbacks, processing, images # pylint: disable=import-error
# import actual nudenet module relative to extension root
from scripts.nudenet import nudenet # pylint: disable=wrong-import-order
from scripts.nudenet import langdetect # pylint: disable=wrong-import-order
from scripts.nudenet import imageguard # pylint: disable=wrong-import-order
from scripts.nudenet import bannedwords # pylint: disable=wrong-import-order
from modules import scripts, scripts_postprocessing, processing, images
from scripts.nudenet import nudenet # pylint: disable=no-name-in-module
from scripts.nudenet import langdetect # pylint: disable=no-name-in-module
from scripts.nudenet import imageguard # pylint: disable=no-name-in-module
from scripts.nudenet import bannedwords # pylint: disable=no-name-in-module
# main ui
@@ -80,7 +77,7 @@ def process(
info = processing.create_infotext(p)
images.save_image(nudes.output, path=p.outpath_samples, seed=p.seed, prompt=p.prompt, info=info, p=p, suffix="-censored")
meta = '; '.join([f'{d["label"]}:{d["score"]}' for d in nudes.detections]) # add all metadata
nsfw = any([d["label"] in nudenet.nsfw for d in nudes.detections]) # noqa:C419
nsfw = any([d["label"] in nudenet.nsfw for d in nudes.detections]) # noqa:C419 # pylint: disable=use-a-generator
if metadata and p is not None:
p.extra_generation_params["NudeNet"] = meta
p.extra_generation_params["NSFW"] = nsfw