yoinked commited on
Commit
791f2ed
1 Parent(s): 0266f59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
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
  )