diff options
Diffstat (limited to 'scripts')
-rw-r--r-- | scripts/custom_code.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/scripts/custom_code.py b/scripts/custom_code.py index 5694f2dd..a9b10c09 100644 --- a/scripts/custom_code.py +++ b/scripts/custom_code.py @@ -4,8 +4,8 @@ import gradio as gr from modules.processing import Processed
from modules.shared import opts, cmd_opts, state
-
class Script(scripts.Script):
+
def title(self):
return "Custom code"
@@ -18,6 +18,7 @@ class Script(scripts.Script): return [code]
+
def run(self, p, code):
assert cmd_opts.allow_code, '--allow-code option must be enabled'
@@ -37,4 +38,5 @@ class Script(scripts.Script): exec(compiled, module.__dict__)
return Processed(p, *display_result_data)
-
+
+
\ No newline at end of file |