feat(dependencies): add image tagging script && update torch dependency

2025-02-21 15:06:01 +09:00
parent 92ab8bcbcc
commit 458a0d9410
2 changed files with 12 additions and 2 deletions
--- a/a.py
+++ b/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)