version: v0.12.0

a.py

@@ -0,0 +1,10 @@
+from PIL import Image
+
+from wdtagger import Tagger
+
+if __name__ == "__main__":
+    tagger = Tagger()
+    images = [
+        Image.open("./tests/images/赤松楓.9d64b955.jpeg"),
+    ]
+    results = tagger.tag(images)

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wdtagger"
-version = "0.11.2"
+version = "0.12.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"
@@ -17,8 +17,8 @@ dev = ["pytest>=8.3.3", "pytest-benchmark>=5.1.0", "ruff>=0.8.0"]
 
 
 [project.optional-dependencies]
-cpu = ["torch>=2.5.1", "torchvision>=0.20.1", "timm>=1.0.11"]
-gpu = ["torch>=2.5.1", "torchvision>=0.20.1", "timm>=1.0.11"]
+cpu = ["torch>=2.0.0", "torchvision>=0.20.1", "timm>=1.0.11"]
+gpu = ["torch>=2.0.0", "torchvision>=0.20.1", "timm>=1.0.11"]
 
 [tool.ruff]
 line-length = 140