Update app.py

app.py

@@ -12,12 +12,20 @@ DESCRIPTION = "Demo for the WaifuDiffusion tagger models"
 
 HF_TOKEN = os.environ.get("HF_TOKEN", "")
 
-# Model Repositories
 VIT_MODEL_DSV3_REPO = "SmilingWolf/wd-vit-tagger-v3"
-
 MODEL_FILENAME = "model.onnx"
 LABEL_FILENAME = "selected_tags.csv"
 
+PREDEFINED_FILTER_TAGS = [
+    "loli", "oppai_loli", "onee-shota", "incest", "furry", "furry_female", "shota",
+    "male_focus", "signature", "otoko_no_ko", "minigirl", "patreon_username", "babydoll",
+    "monochrome", "happy_birthday", "happy_new_year", "thought_bubble", "greyscale",
+    "speech_bubble", "english_text", "copyright_name", "twitter_username",
+    "patreon username", "patreon logo", "cover", "content_rating", "cover_page",
+    "doujin_cover", "sex", "artist_name", "watermark", "censored", "bar_censor",
+    "blank_censor", "blur_censor", "light_censor", "mosaic_censoring"
+]
+
 def parse_args():
     parser = argparse.ArgumentParser()
     parser.add_argument("--score-slider-step", type=float, default=0.05)
@@ -154,30 +162,14 @@ with gr.Blocks(title=TITLE) as demo:
     gr.Markdown(f"<h1 style='text-align: center;'>{TITLE}</h1>")
     gr.Markdown(DESCRIPTION)
 
-    with gr.Row():
-        with gr.Column():
-            image_files = gr.File(file_types=["image"], label="Upload Images", file_count="multiple")
-
-            with gr.Accordion("Advanced Settings", open=False):
-                model_repo = gr.Dropdown([VIT_MODEL_DSV3_REPO], value=VIT_MODEL_DSV3_REPO, label="Select Model")
-                general_thresh = gr.Slider(0, 1, step=args.score_slider_step, value=args.score_general_threshold, label="General Tags Threshold")
-                character_thresh = gr.Slider(0, 1, step=args.score_slider_step, value=args.score_character_threshold, label="Character Tags Threshold")
-                filter_tags = gr.Textbox(label="Filter Tags (comma-separated)", lines=3)
-
-            submit = gr.Button(value="Process Images", variant="primary")
-
-        with gr.Column():
-            output = gr.Textbox(label="Output", lines=10)
-
-    with gr.Accordion("Tag Replacements", open=False):
-        replacement_rules_text = gr.Textbox(label="Replacement Rules", placeholder="e.g., 1boy -> 1girl", lines=5)
+    with gr.Accordion("Settings", open=False):
+        filter_tags = gr.Textbox(value=", ".join(PREDEFINED_FILTER_TAGS), label="Filter Tags (comma-separated)", lines=3)
+        replacement_rules_text = gr.Textbox(label="Replacement Rules", value="1boy -> 1girl", lines=5)
+        fallback_rules_text = gr.Textbox(label="Fallback Rules", value="sad, happy -> smile", lines=5)
 
-    with gr.Accordion("Fallback Rules", open=False):
-        fallback_rules_text = gr.Textbox(label="Fallback Rules", placeholder="e.g., sad, happy -> smile", lines=5)
+    submit = gr.Button(value="Process Images")
+    output = gr.Textbox(label="Output", lines=10)
 
-    submit.click(process_images, 
-                 inputs=[image_files, model_repo, general_thresh, character_thresh, filter_tags, replacement_rules_text, fallback_rules_text], 
-                 outputs=output)
+    submit.click(process_images, inputs=[[], VIT_MODEL_DSV3_REPO, args.score_general_threshold, args.score_character_threshold, filter_tags, replacement_rules_text, fallback_rules_text], outputs=output)
 
-demo.queue(max_size=10)
 demo.launch()