Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -116,6 +116,9 @@ class WaifuDiffusionInterrogator:
|
|
| 116 |
|
| 117 |
WAIFU_MODELS: Mapping[str, WaifuDiffusionInterrogator] = {
|
| 118 |
'wd14-vit': WaifuDiffusionInterrogator(),
|
|
|
|
|
|
|
|
|
|
| 119 |
}
|
| 120 |
RE_SPECIAL = re.compile(r'([\\()])')
|
| 121 |
|
|
|
|
| 116 |
|
| 117 |
WAIFU_MODELS: Mapping[str, WaifuDiffusionInterrogator] = {
|
| 118 |
'wd14-vit': WaifuDiffusionInterrogator(),
|
| 119 |
+
'wd14-convnext': WaifuDiffusionInterrogator(
|
| 120 |
+
repo='SmilingWolf/wd-v1-4-convnext-tagger'
|
| 121 |
+
),
|
| 122 |
}
|
| 123 |
RE_SPECIAL = re.compile(r'([\\()])')
|
| 124 |
|