Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -124,7 +124,16 @@ WAIFU_MODELS: Mapping[str, WaifuDiffusionInterrogator] = {
|
|
| 124 |
),
|
| 125 |
'chen-moat2': WaifuDiffusionInterrogator(
|
| 126 |
repo='SmilingWolf/wd-v1-4-moat-tagger-v2'
|
| 127 |
-
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 128 |
}
|
| 129 |
RE_SPECIAL = re.compile(r'([\\()])')
|
| 130 |
|
|
|
|
| 124 |
),
|
| 125 |
'chen-moat2': WaifuDiffusionInterrogator(
|
| 126 |
repo='SmilingWolf/wd-v1-4-moat-tagger-v2'
|
| 127 |
+
),
|
| 128 |
+
'chen-convnext3': WaifuDiffusionInterrogator(
|
| 129 |
+
repo='SmilingWolf/wd-convnext-tagger-v3'
|
| 130 |
+
),
|
| 131 |
+
'chen-vit3': WaifuDiffusionInterrogator(
|
| 132 |
+
repo='SmilingWolf/wd-vit-tagger-v3'
|
| 133 |
+
),
|
| 134 |
+
'chen-swinv3': WaifuDiffusionInterrogator(
|
| 135 |
+
repo='SmilingWolf/wd-swinv2-tagger-v3'
|
| 136 |
+
),
|
| 137 |
}
|
| 138 |
RE_SPECIAL = re.compile(r'([\\()])')
|
| 139 |
|