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