Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -118,24 +118,30 @@ WAIFU_MODELS: Mapping[str, WaifuDiffusionInterrogator] = {
|
|
118 |
'chen-convnext': WaifuDiffusionInterrogator(
|
119 |
repo='SmilingWolf/wd-v1-4-convnext-tagger'
|
120 |
),
|
121 |
-
'chen-convnext2': WaifuDiffusionInterrogator(
|
122 |
repo="SmilingWolf/wd-v1-4-convnextv2-tagger-v2"
|
123 |
),
|
124 |
-
'chen-
|
125 |
repo='SmilingWolf/wd-v1-4-swinv2-tagger-v2'
|
126 |
),
|
127 |
-
'chen-
|
128 |
repo='SmilingWolf/wd-v1-4-moat-tagger-v2'
|
129 |
),
|
130 |
-
'chen-
|
131 |
repo='SmilingWolf/wd-convnext-tagger-v3'
|
132 |
),
|
133 |
-
'chen-
|
134 |
repo='SmilingWolf/wd-vit-tagger-v3'
|
135 |
),
|
136 |
'chen-swinv3': WaifuDiffusionInterrogator(
|
137 |
repo='SmilingWolf/wd-swinv2-tagger-v3'
|
138 |
),
|
|
|
|
|
|
|
|
|
|
|
|
|
139 |
}
|
140 |
RE_SPECIAL = re.compile(r'([\\()])')
|
141 |
|
|
|
118 |
'chen-convnext': WaifuDiffusionInterrogator(
|
119 |
repo='SmilingWolf/wd-v1-4-convnext-tagger'
|
120 |
),
|
121 |
+
'chen-convnext2-v2': WaifuDiffusionInterrogator(
|
122 |
repo="SmilingWolf/wd-v1-4-convnextv2-tagger-v2"
|
123 |
),
|
124 |
+
'chen-swin2': WaifuDiffusionInterrogator(
|
125 |
repo='SmilingWolf/wd-v1-4-swinv2-tagger-v2'
|
126 |
),
|
127 |
+
'chen-moatv2': WaifuDiffusionInterrogator(
|
128 |
repo='SmilingWolf/wd-v1-4-moat-tagger-v2'
|
129 |
),
|
130 |
+
'chen-convnextv3': WaifuDiffusionInterrogator(
|
131 |
repo='SmilingWolf/wd-convnext-tagger-v3'
|
132 |
),
|
133 |
+
'chen-vitv3': WaifuDiffusionInterrogator(
|
134 |
repo='SmilingWolf/wd-vit-tagger-v3'
|
135 |
),
|
136 |
'chen-swinv3': WaifuDiffusionInterrogator(
|
137 |
repo='SmilingWolf/wd-swinv2-tagger-v3'
|
138 |
),
|
139 |
+
'chen-vit-largev3': WaifuDiffusionInterrogator(
|
140 |
+
repo='SmilingWolf/wd-vit-large-tagger-v3'
|
141 |
+
),
|
142 |
+
'chen-evangelion': WaifuDiffusionInterrogator(
|
143 |
+
repo='SmilingWolf/wd-eva02-large-tagger-v3'
|
144 |
+
),
|
145 |
}
|
146 |
RE_SPECIAL = re.compile(r'([\\()])')
|
147 |
|