1580158
/
automatic
mirror of https://github.com/vladmandic/automatic
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Revert import format change 

Not sure why, but it was causing errors
pull/4722/head
awsr 2026-03-29 02:04:22 -07:00
parent 59b9ca50ee
commit 95dadab5c3
No known key found for this signature in database
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
scripts/nudenet_ext.py
View File

@ -1,7 +1,6 @@
import time import time
import gradio as gr import gradio as gr
from modules import scripts, processing, images from modules import scripts, scripts_postprocessing, processing, images
from modules.scripts_postprocessing import PostprocessedImage, ScriptPostprocessing
from scripts.nudenet import nudenet # pylint: disable=no-name-in-module 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 langdetect # pylint: disable=no-name-in-module
from scripts.nudenet import imageguard # pylint: disable=no-name-in-module from scripts.nudenet import imageguard # pylint: disable=no-name-in-module
@ -51,7 +50,7 @@ def create_ui(accordion=True):
# main processing used in both modes # main processing used in both modes
def process( def process(
p: processing.StableDiffusionProcessing | None = None, p: processing.StableDiffusionProcessing | None = None,
pp: scripts.PostprocessImageArgs | PostprocessedImage | None = None, pp: scripts.PostprocessImageArgs | scripts_postprocessing.PostprocessedImage | None = None,
enabled=True, enabled=True,
lang=False, lang=False,
policy=False, policy=False,
@ -86,7 +85,7 @@ def process(
if metadata and p is not None: if metadata and p is not None:
p.extra_generation_params["NudeNet"] = meta p.extra_generation_params["NudeNet"] = meta
p.extra_generation_params["NSFW"] = nsfw p.extra_generation_params["NSFW"] = nsfw
if metadata and isinstance(pp, PostprocessedImage): if metadata and isinstance(pp, scripts_postprocessing.PostprocessedImage):
pp.info['NudeNet'] = meta pp.info['NudeNet'] = meta
pp.info['NSFW'] = nsfw pp.info['NSFW'] = nsfw
log.debug(f'NudeNet detect: {dct} nsfw={nsfw} time={(t1 - t0):.2f}') log.debug(f'NudeNet detect: {dct} nsfw={nsfw} time={(t1 - t0):.2f}')
@ -119,7 +118,7 @@ def process(
if metadata and p is not None: if metadata and p is not None:
p.extra_generation_params["Rating"] = res.get('rating', 'N/A') p.extra_generation_params["Rating"] = res.get('rating', 'N/A')
p.extra_generation_params["Category"] = res.get('category', 'N/A') p.extra_generation_params["Category"] = res.get('category', 'N/A')
if metadata and isinstance(pp, PostprocessedImage): if metadata and isinstance(pp, scripts_postprocessing.PostprocessedImage):
pp.info["Rating"] = res.get('rating', 'N/A') pp.info["Rating"] = res.get('rating', 'N/A')
pp.info["Category"] = res.get('category', 'N/A') pp.info["Category"] = res.get('category', 'N/A')
@ -149,7 +148,7 @@ class ScriptNudeNet(scripts.Script):
# defines postprocessing script for dual-mode usage # defines postprocessing script for dual-mode usage
class ScriptPostprocessingNudeNet(ScriptPostprocessing): class ScriptPostprocessingNudeNet(scripts_postprocessing.ScriptPostprocessing):
name = 'NudeNet' name = 'NudeNet'
order = 10000 order = 10000
@ -159,5 +158,5 @@ class ScriptPostprocessingNudeNet(ScriptPostprocessing):
return { 'enabled': enabled, 'lang': lang, 'policy': policy, 'banned': banned, 'metadata': metadata, 'copy': copy, 'score': score, 'blocks': blocks, 'censor': censor, 'method': method, 'overlay': overlay, 'allowed': allowed, 'alphabet': alphabet, 'words': words} return { 'enabled': enabled, 'lang': lang, 'policy': policy, 'banned': banned, 'metadata': metadata, 'copy': copy, 'score': score, 'blocks': blocks, 'censor': censor, 'method': method, 'overlay': overlay, 'allowed': allowed, 'alphabet': alphabet, 'words': words}
# triggered by callback # triggered by callback
def process(self, pp: PostprocessedImage, enabled, lang, policy, banned, metadata, copy, score, blocks, censor, method, overlay, allowed, alphabet, words): # pylint: disable=arguments-differ def process(self, pp: scripts_postprocessing.PostprocessedImage, enabled, lang, policy, banned, metadata, copy, score, blocks, censor, method, overlay, allowed, alphabet, words): # pylint: disable=arguments-differ
process(None, pp, enabled, lang, policy, banned, metadata, copy, score, blocks, censor, method, overlay, allowed, alphabet, words) process(None, pp, enabled, lang, policy, banned, metadata, copy, score, blocks, censor, method, overlay, allowed, alphabet, words)