add state to interrogate

d8d206c1 · Vladimir Mandic · GitHub · 192ddc04 · d8d206c1
Commit d8d206c1 authored Jan 03, 2023 by Vladimir Mandic Committed by GitHub Jan 03, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

modules/interrogate.py modules/interrogate.py +3 -1

No files found.
--- a/modules/interrogate.py
+++ b/modules/interrogate.py
@@ -136,7 +136,8 @@ class InterrogateModels:

    def interrogate(self, pil_image):
        res = ""
-
+        shared.state.begin()
+        shared.state.job = 'interrogate'
        try:

            if shared.cmd_opts.lowvram or shared.cmd_opts.medvram:
@@ -177,5 +178,6 @@ class InterrogateModels:
            res += "<error>"

        self.unload()
+        shared.state.end()

        return res