FrostIce commited on
Commit
2ab74b1
·
verified ·
1 Parent(s): 5d846dd

Update app.py

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