diff --git a/pyproject.toml b/pyproject.toml index 185e087..91c9cb1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "wdtagger" -version = "0.13.2" +version = "0.14.0" description = "A simple and easy-to-use wrapper for the tagger model created by [SmilingWolf](https://github.com/SmilingWolf) which is specifically designed for tagging anime illustrations." authors = [{ name = "Jianqi Pan", email = "jannchie@gmail.com" }] readme = "README.md" diff --git a/temp.py b/temp.py new file mode 100644 index 0000000..e54a5f8 --- /dev/null +++ b/temp.py @@ -0,0 +1,7 @@ +from wdtagger import Tagger + +tagger = Tagger() + + +resp = tagger.tag("./tests/images/GkZFQ-_WwAAeNlO.jpg", character_threshold=0.85, general_threshold=0.35) +print(resp.character_tags) diff --git a/tests/images/GkZFQ-_WwAAeNlO.jpg b/tests/images/GkZFQ-_WwAAeNlO.jpg new file mode 100644 index 0000000..99d5f11 Binary files /dev/null and b/tests/images/GkZFQ-_WwAAeNlO.jpg differ