make CLIP interrogate ranks output sane values

1 files changed, 3 insertions, 3 deletions

diff --git a/modules/interrogate.py b/modules/interrogate.py
index d85d7dcc..64b91eb4 100644
--- a/modules/interrogate.py
+++ b/modules/interrogate.py
@@ -123,7 +123,7 @@ class InterrogateModels:
 

         return caption[0]

 

-    def interrogate(self, pil_image, include_ranks=False):

+    def interrogate(self, pil_image):

         res = None

 

         try:

@@ -156,8 +156,8 @@ class InterrogateModels:
                 for name, topn, items in self.categories:

                     matches = self.rank(image_features, items, top_count=topn)

                     for match, score in matches:

-                        if include_ranks:

-                            res += f", ({match}:{score})"

+                        if shared.opts.interrogate_return_ranks:

+                            res += f", ({match}:{score/100:.3f})"

                         else:

                             res += ", " + match