Update app.py

app.py

@@ -115,8 +115,8 @@ class WaifuDiffusionInterrogator:
 
 
 WAIFU_MODELS: Mapping[str, WaifuDiffusionInterrogator] = {
-    'wd14-vit': WaifuDiffusionInterrogator(),
-    'wd14-convnext': WaifuDiffusionInterrogator(
+    'chen-vit': WaifuDiffusionInterrogator(),
+    'chen-convnext': WaifuDiffusionInterrogator(
         repo='SmilingWolf/wd-v1-4-convnext-tagger'
     ),
 }
@@ -124,7 +124,7 @@ RE_SPECIAL = re.compile(r'([\\()])')
 
 
 def image_to_wd14_tags(image: Image.Image, model_name: str, threshold: float,
-                       use_spaces: bool, use_escape: bool, include_ranks: bool, score_descend: bool) \
+                       use_spaces: bool, use_escape: bool, include_ranks=False, score_descend=True) \
         -> Tuple[Mapping[str, float], str, Mapping[str, float]]:
     model = WAIFU_MODELS[model_name]
     ratings, tags = model.interrogate(image)
@@ -156,29 +156,27 @@ if __name__ == '__main__':
     with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
-                gr_input_image = gr.Image(type='pil', label='Original Chen')
+                gr_input_image = gr.Image(type='pil', label='Chen Chen')
                 with gr.Row():
-                    gr_model = gr.Radio(list(WAIFU_MODELS.keys()), value='wd14-vit', label='Waifu Model')
+                    gr_model = gr.Radio(list(WAIFU_MODELS.keys()), value='chen-vit', 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=False, label='Use Space Instead Of _')
-                    gr_escape = gr.Checkbox(value=True, label='Use Text Escape')
-                    gr_confidence = gr.Checkbox(value=False, label='Keep Confidences')
-                    gr_order = gr.Checkbox(value=True, label='Descend By Confidence')
+                    gr_space = gr.Checkbox(value=True, label='Chen " " Chen Chen "_"')
+                    gr_escape = gr.Checkbox(value=True, label='Chen Text Escape')
 
-                gr_btn_submit = gr.Button(value='Tagging', variant='primary')
+                gr_btn_submit = gr.Button(value='橙', variant='primary')
 
             with gr.Column():
-                gr_ratings = gr.Label(label='Ratings')
+                gr_ratings = gr.Label(label='橙 橙')
                 with gr.Tabs():
-                    with gr.Tab("Tags"):
-                        gr_tags = gr.Label(label='Tags')
-                    with gr.Tab("Exported Text"):
-                        gr_output_text = gr.TextArea(label='Exported Text')
+                    with gr.Tab("Chens"):
+                        gr_tags = gr.Label(label='Chens')
+                    with gr.Tab("Chen Text"):
+                        gr_output_text = gr.TextArea(label='Chen Text')
 
         gr_btn_submit.click(
             image_to_wd14_tags,
-            inputs=[gr_input_image, gr_model, gr_threshold, gr_space, gr_escape, gr_confidence, gr_order],
+            inputs=[gr_input_image, gr_model, gr_threshold, gr_space, gr_escape],
             outputs=[gr_ratings, gr_output_text, gr_tags],
         )
     demo.queue(os.cpu_count()).launch()