Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -126,6 +126,12 @@ WAIFU_MODELS: Mapping[str, WaifuDiffusionInterrogator] = {
|
|
126 |
'chen-convnext': WaifuDiffusionInterrogator(
|
127 |
repo='SmilingWolf/wd-v1-4-convnext-tagger'
|
128 |
),
|
|
|
|
|
|
|
|
|
|
|
|
|
129 |
'chen-moat2': WaifuDiffusionInterrogator(
|
130 |
repo='SmilingWolf/wd-v1-4-moat-tagger-v2'
|
131 |
)
|
|
|
126 |
'chen-convnext': WaifuDiffusionInterrogator(
|
127 |
repo='SmilingWolf/wd-v1-4-convnext-tagger'
|
128 |
),
|
129 |
+
'chen-convnext2': WaifuDiffusionInterrogator(
|
130 |
+
repo="SmilingWolf/wd-v1-4-convnextv2-tagger-v2"
|
131 |
+
),
|
132 |
+
'chen-swinv2': WaifuDiffusionInterrogator(
|
133 |
+
repo='SmilingWolf/wd-v1-4-swinv2-tagger-v2'
|
134 |
+
),
|
135 |
'chen-moat2': WaifuDiffusionInterrogator(
|
136 |
repo='SmilingWolf/wd-v1-4-moat-tagger-v2'
|
137 |
)
|