add pyproject and reqs

README.md

@@ -9,7 +9,7 @@ app_file: app.py
 pinned: true
 license: MIT
 preload_from_hub:
-  - SmilingWolf/wd-vit-tagger-v3 model.onnx,selected_tags.csv
+  - SmilingWolf/wd-vit-tagger-v3 model.safetensors,config.json,sw_jax_cv_config.json,selected_tags.csv
 models:
   - SmilingWolf/wd-vit-tagger-v3
 ---

pyproject.toml

@@ -0,0 +1,94 @@
+[project]
+name = "wd-tagger-heatmap"
+version = "0.1.0"
+authors = [
+  { name = "Andi Powers-Holmes", email = "[email protected]" },
+]
+maintainers = [
+  { name = "Andi Powers-Holmes", email = "[email protected]" },
+]
+description = "wd-tagger-heatmap Gradio Space"
+readme = "README.md"
+requires-python = ">=3.9, <3.11"
+keywords = [
+  "deep-learning",
+  "machine-learning",
+  "pytorch",
+]
+license = { file = "LICENSE.md" }
+classifiers = [
+  "Programming Language :: Python :: 3",
+  "License :: OSI Approved :: MIT License",
+]
+dependencies = [
+  "colorcet",
+  "diffusers",
+  "gradio >=4.25.0, < 5.0.0",
+  "hf-transfer",
+  "huggingface-hub >= 0.14.0",
+  "matplotlib",
+  "numpy >= 1.23.5",
+  "opencv-contrib-python-headless",
+  "pandas >= 2.0.0",
+  "Pillow >= 9.5.0",
+  "rich",
+  "timm @ git+https://github.com/huggingface/pytorch-image-models@main#egg=timm",
+  "tokenizers",
+  "torch >= 2.1.0",
+  "torchvision",
+  "transformers",
+]
+
+[project.urls]
+Repository = "https://huggingface.co/spaces/neggles/wd-tagger-heatmap"
+
+[project.optional-dependencies]
+dev = [
+  "ruff >=0.0.289",
+  "setuptools-scm >= 8.0.0",
+  "pre-commit >= 3.0.0",     # remember to run `pre-commit install` after installing
+  "tabulate >= 0.8.9",       # for inductor log prettyprinting
+]
+all = [
+  "wd-tagger-heatmap-space[dev]",
+]
+
+[build-system]
+build-backend = "setuptools.build_meta"
+requires = ["setuptools>=64", "wheel"]
+
+[tool.setuptools.packages.find]
+namespaces = true
+where = ["."]
+include = ["wd-tagger-heatmap"]
+
+
+[tool.ruff]
+line-length = 110
+target-version = "py310"
+extend-exclude = ["/usr/lib/*"]
+
+[tool.ruff.lint]
+ignore = [
+  "F841", # local variable assigned but never used
+  "F842", # local variable annotated but never used
+  "E501", # line too long - will be fixed in format
+]
+
+[tool.ruff.format]
+quote-style = "double"
+indent-style = "space"
+line-ending = "auto"
+skip-magic-trailing-comma = false
+docstring-code-format = true
+
+[tool.ruff.lint.isort]
+combine-as-imports = true
+force-wrap-aliases = true
+known-first-party = ["tagger"]
+
+
+[tool.pyright]
+include = ["src/**"]
+exclude = ["/usr/lib/**"]
+stubPath = "./typings"

requirements.txt

@@ -0,0 +1,16 @@
+colorcet
+diffusers
+gradio >=4.25.0, < 5.0.0
+hf-transfer
+huggingface-hub >= 0.14.0
+matplotlib
+numpy >= 1.23.5
+opencv-contrib-python-headless
+pandas >= 2.0.0
+Pillow >= 9.5.0
+rich
+timm @ git+https://github.com/huggingface/pytorch-image-models@main#egg=timm
+tokenizers
+torch >= 2.1.0
+torchvision
+transformers