Merge pull request #188 from C-BP/main

docs:add Tagger API example

README.md

@@ -506,6 +506,16 @@ semantic_seg_result = segment.sam_and_semantic_seg_with_cat_id(
 )
 ```
 
+### Extension support - Tagger (contributed by C-BP)
+
+```python
+# https://github.com/Akegarasu/sd-webui-wd14-tagger
+
+tagger = webuiapi.TaggerInterface(api)
+result = tagger.tagger_interrogate(image)
+print(result)
+# {"caption": {"additionalProp1":0.9,"additionalProp2": 0.8,"additionalProp3": 0.7}}
+```
 ### Extension support - ADetailer (contributed by tomj2ee and davidmartinrius)
 #### txt2img with ADetailer
 ```

webuiapi/webuiapi.py

@@ -2009,14 +2009,14 @@ class TaggerInterface:
     def __init__(self, webuiapi: WebUIApi):
         self.api = webuiapi
 
-    def tagger_interrogate(self, image, model="wd14-vit-v2-git", threshold=0.0, use_async=False):
+    def tagger_interrogate(self, image, model="wd14-vit-v2-git", threshold=0.3, use_async=False):
         """
         Interrogates the tagger model with the provided image and parameters.
 
         Args:
             image (Image.Image or str): The image to be interrogated. Can be a PIL Image object or a base64 encoded string.
             model (str, optional): The model to use for interrogation. Defaults to "wd14-vit-v2-git".
-            threshold (float, optional): The threshold value for the model. Defaults to 0.
+            threshold (float, optional): The threshold value for the model. Defaults to 0.3
             use_async (bool, optional): Whether to use asynchronous processing. Defaults to False.
 
         Returns: