From 458a0d941008ac7e4b30310aa64c909e7cee1171 Mon Sep 17 00:00:00 2001
From: Jianqi Pan <jannchie@gmail.com>
Date: Fri, 21 Feb 2025 15:06:01 +0900
Subject: [PATCH] feat(dependencies): add image tagging script && update torch
 dependency

---
 a.py           | 10 ++++++++++
 pyproject.toml |  4 ++--
 2 files changed, 12 insertions(+), 2 deletions(-)
 create mode 100644 a.py

diff --git a/a.py b/a.py
new file mode 100644
index 0000000..d63ab8e
--- /dev/null
+++ 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)
diff --git a/pyproject.toml b/pyproject.toml
index 12966ad..d0f048f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -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