yoinked commited on
Commit
485a97c
·
verified ·
1 Parent(s): 598e930

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -5
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-swinv2': WaifuDiffusionInterrogator(
125
  repo='SmilingWolf/wd-v1-4-swinv2-tagger-v2'
126
  ),
127
- 'chen-moat2': WaifuDiffusionInterrogator(
128
  repo='SmilingWolf/wd-v1-4-moat-tagger-v2'
129
  ),
130
- 'chen-convnext3': WaifuDiffusionInterrogator(
131
  repo='SmilingWolf/wd-convnext-tagger-v3'
132
  ),
133
- 'chen-vit3': WaifuDiffusionInterrogator(
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