Update app.py

app.py

@@ -141,6 +141,12 @@ WAIFU_MODELS: Mapping[str, WaifuDiffusionInterrogator] = {
     'chen-evangelion': WaifuDiffusionInterrogator(
         repo='SmilingWolf/wd-eva02-large-tagger-v3'
     ),
+    'chenkaku-evangelion': WaifuDiffusionInterrogator(
+        repo='deepghs/idolsankaku-eva02-large-tagger-v1'
+    ),
+    'chenkaku-swinv2': WaifuDiffusionInterrogator(
+        repo='deepghs/idolsankaku-swinv2-tagger-v1'
+    ),
 }
 RE_SPECIAL = re.compile(r'([\\()])')
 
@@ -185,7 +191,7 @@ if __name__ == '__main__':
             with gr.Column():
                 gr_input_image = gr.Image(type='pil', label='Chen Chen', sources=['upload', 'clipboard'])
                 with gr.Row():
-                    gr_model = gr.Radio(list(WAIFU_MODELS.keys()), value='chen-moat2', label='Chen')
+                    gr_model = gr.Radio(list(WAIFU_MODELS.keys()), value='chen-vit-largev3', label='Chen')
                     gr_threshold = gr.Slider(0.0, 1.0, 0.5, label='Chen Chen Chen Chen Chen')
                 with gr.Row():
                     gr_space = gr.Checkbox(value=True, label='Use DashSpace')