da_nsfw_checker

Running

yoinked commited on Feb 26

Commit

0266f59

•

1 Parent(s): 0f5a1c8

moattest

Files changed (1) hide show

app.py CHANGED Viewed

@@ -126,6 +126,9 @@ WAIFU_MODELS: Mapping[str, WaifuDiffusionInterrogator] = {
     'chen-convnext': WaifuDiffusionInterrogator(
         repo='SmilingWolf/wd-v1-4-convnext-tagger'
     ),
 }
 RE_SPECIAL = re.compile(r'([\\()])')

     'chen-convnext': WaifuDiffusionInterrogator(
         repo='SmilingWolf/wd-v1-4-convnext-tagger'
     ),
+    'chen-moat2': WaifuDiffusionInterrogator(
+        repo='SmilingWolf/wd-v1-4-moat-tagger-v2'
+    )
 }
 RE_SPECIAL = re.compile(r'([\\()])')