add state to interrogate

This commit is contained in:
Vladimir Mandic
2023-01-03 11:01:04 -05:00
committed by GitHub
parent 192ddc04d6
commit d8d206c168

View File

@@ -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