Update app.py

app.py

@@ -166,16 +166,30 @@ if __name__ == '__main__':
                 gr_btn_submit = gr.Button(value='Chening', variant='primary')
 
             with gr.Column():
-                gr_ratings = gr.Label(label='Chen?')
+                gr_input_image = gr.Image(type='pil', label='Original Chen')
+                with gr.Row():
+                    gr_model = gr.Radio(list(WAIFU_MODELS.keys()), value='wd14-vit', label='Waifu Model')
+                    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_btn_submit = gr.Button(value='Tagging', variant='primary')
+
+            with gr.Column():
+                gr_ratings = gr.Label(label='Ratings')
                 with gr.Tabs():
-                    with gr.Tab("Chen"):
-                        gr_tags = gr.Label(label='Chen')
-                    with gr.Tab("Exported Chen"):
-                        gr_output_text = gr.TextArea(label='Exported Chen')
+                    with gr.Tab("Tags"):
+                        gr_tags = gr.Label(label='Tags')
+                    with gr.Tab("Exported Text"):
+                        gr_output_text = gr.TextArea(label='Exported Text')
 
         gr_btn_submit.click(
             image_to_wd14_tags,
-            inputs=[gr_input_image, gr_model, gr_threshold, gr_space, True, True, True],
+            inputs=[gr_input_image, gr_model, gr_threshold, gr_space, gr_escape, gr_confidence, gr_order],
             outputs=[gr_ratings, gr_output_text, gr_tags],
         )
+
     demo.queue(os.cpu_count()).launch()