update model ids

README.md

@@ -40,7 +40,7 @@ cp config.example.json config.json
 env/bin/python3 ./main.py fight
 
 # score mode -- crank the final numbers, generates graphs under plots/ folder
-env/bin/python3 ./main.py fight
+env/bin/python3 ./main.py scores
 
 # keep in mind that you can download more images, run fight mode, and then
 # run score mode! the commands are aware of work that's been already done and

main.py

@@ -24,13 +24,17 @@ from dataclasses import dataclass, field
 log = logging.getLogger(__name__)
 
 
+# from https://github.com/picobyte/stable-diffusion-webui-wd14-tagger/blob/49cfdcd6be77086e6cc0d26b5ec26605b702be6e/tagger/utils.py#L15C1-L58C1
 DEFAULTS = [
-    "wd14-vit-v2-git",
-    "wd14-vit",
-    "wd14-swinv2-v2-git",
-    "wd14-convnextv2-v2-git",
-    "wd14-convnext-v2-git",
-    "wd14-convnext",
+    "wd14-vit.v1",
+    "wd14-vit.v2",
+    "wd14-convnext.v1",
+    "wd14-convnext.v2",
+    "wd14-convnextv2.v1",
+    "wd14-swinv2-v1",
+    "wd-v1-4-moat-tagger.v2",
+    "mld-caformer.dec-5-97527",
+    # broken model: "mld-tresnetd.6-30000",
 ]
 
 

requirements.txt

@@ -1,8 +1,7 @@
 aiosqlite==0.19.0
 aiohttp==3.8.4
 aiolimiter>1.1.0<2.0
-
-
+aiofiles==23.1.0
 plotly>5.15.0<6.0
 pandas==2.0.2
 kaleido==0.2.1