From 8bce746fa22b7ba80474755c1f7da736e3ba9baa Mon Sep 17 00:00:00 2001 From: vladmandic Date: Wed, 8 Apr 2026 08:39:42 +0200 Subject: [PATCH] lint Signed-off-by: vladmandic --- scripts/nudenet/imageguard.py | 6 ++++-- scripts/nudenet_ext.py | 6 +++--- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/scripts/nudenet/imageguard.py b/scripts/nudenet/imageguard.py index c92dd43ee..aa3a881be 100644 --- a/scripts/nudenet/imageguard.py +++ b/scripts/nudenet/imageguard.py @@ -1,11 +1,11 @@ from __future__ import annotations - -from modules.logger import log from typing import TYPE_CHECKING +from modules.logger import log if TYPE_CHECKING: from transformers import AutoProcessor, LlavaOnevisionForConditionalGeneration + repo_id = 'AIML-TUDA/LlavaGuard-v1.2-0.5B-OV-hf' policy_template = """Provide a safety assessment for the content provided by the user. Determine compliance or violation against our safety policy by reviewing the following policy categories: Hate: @@ -96,6 +96,8 @@ To provide your assessment use the following json template for each category: "rationale": str, }. """ + + model: LlavaOnevisionForConditionalGeneration | None = None processor: AutoProcessor | None = None diff --git a/scripts/nudenet_ext.py b/scripts/nudenet_ext.py index 45208832c..41a60b162 100644 --- a/scripts/nudenet_ext.py +++ b/scripts/nudenet_ext.py @@ -56,7 +56,7 @@ def process( policy=False, banned=False, metadata=True, - copy=False, # Compatability + copy=False, # pylint: disable=unused-argument # compatability score=0.2, blocks=3, censor=[], @@ -131,11 +131,11 @@ class ScriptNudeNet(scripts.Script): def title(self): return 'NudeNet' - def show(self, *args, **kwargs): + def show(self, *args, **kwargs): # pylint: disable=unused-argument return scripts.AlwaysVisible # return signature is array of gradio components - def ui(self, *args, **kwargs): + def ui(self, *args, **kwargs): # pylint: disable=unused-argument return create_ui(accordion=True) # triggered by callback