Upload app.py

app.py

@@ -3,65 +3,24 @@ import gradio as gr
 import numpy as np
 
 # DiffuseCraft
-from dc import (
-    infer,
-    _infer,
-    pass_result,
-    get_diffusers_model_list,
-    get_samplers,
-    get_vaes,
-    enable_model_recom_prompt,
-    enable_diffusers_model_detail,
-    get_t2i_model_info,
-    get_all_lora_tupled_list,
-    update_loras,
-    apply_lora_prompt,
-    download_my_lora,
-    search_civitai_lora,
-    select_civitai_lora,
-    search_civitai_lora_json,
-    preset_quality,
-    preset_styles,
-    process_style_prompt,
-)
+from dc import (infer, _infer, pass_result, get_diffusers_model_list, get_samplers,
+    get_vaes, enable_model_recom_prompt, enable_diffusers_model_detail,
+    get_t2i_model_info, get_all_lora_tupled_list, update_loras,
+    apply_lora_prompt, download_my_lora, search_civitai_lora,
+    select_civitai_lora, search_civitai_lora_json,
+    preset_quality, preset_styles, process_style_prompt)
 # Translator
-from llmdolphin import (
-    dolphin_respond_auto,
-    dolphin_parse_simple,
-    get_llm_formats,
-    get_dolphin_model_format,
-    get_dolphin_models,
-    get_dolphin_model_info,
-    select_dolphin_model,
-    select_dolphin_format,
-    get_dolphin_sysprompt,
-)
+from llmdolphin import (dolphin_respond_auto, dolphin_parse_simple,
+    get_llm_formats, get_dolphin_model_format, get_dolphin_models,
+    get_dolphin_model_info, select_dolphin_model, select_dolphin_format, get_dolphin_sysprompt)
 # Tagger
-from tagger.v2 import (
-    v2_upsampling_prompt,
-    V2_ALL_MODELS,
-)
-from tagger.utils import (
-    gradio_copy_text,
-    gradio_copy_prompt,
-    COPY_ACTION_JS,
-    V2_ASPECT_RATIO_OPTIONS,
-    V2_RATING_OPTIONS,
-    V2_LENGTH_OPTIONS,
-    V2_IDENTITY_OPTIONS
-)
-from tagger.tagger import (
-    predict_tags_wd,
-    convert_danbooru_to_e621_prompt,
-    remove_specific_prompt,
-    insert_recom_prompt,
-    compose_prompt_to_copy,
-    translate_prompt,
-    select_random_character,
-)
-from tagger.fl2sd3longcap import (
-    predict_tags_fl2_sd3,
-)
+from tagger.v2 import v2_upsampling_prompt, V2_ALL_MODELS
+from tagger.utils import (gradio_copy_text, gradio_copy_prompt, COPY_ACTION_JS,
+    V2_ASPECT_RATIO_OPTIONS, V2_RATING_OPTIONS, V2_LENGTH_OPTIONS, V2_IDENTITY_OPTIONS)
+from tagger.tagger import (predict_tags_wd, convert_danbooru_to_e621_prompt,
+    remove_specific_prompt, insert_recom_prompt, compose_prompt_to_copy,
+    translate_prompt, select_random_character)
+from tagger.fl2sd3longcap import predict_tags_fl2_sd3
 def description_ui():
     gr.Markdown(
         """
@@ -219,8 +178,8 @@ with gr.Blocks(css=css, fill_width=True, elem_id="container") as demo:
                 
                 vae_model = gr.Dropdown(label="VAE Model", choices=get_vaes(), value=get_vaes()[0])
                 recom_prompt = gr.Checkbox(label="Recommended prompt", value=True)
-                quality_selector = gr.Dropdown(label="Quality Tags Presets", interactive=True, choices=list(preset_quality.keys()), value="None")
-                style_selector = gr.Dropdown(label="Style Preset", interactive=True, choices=list(preset_styles.keys()), value="None")
+                quality_selector = gr.Radio(label="Quality Tag Presets", interactive=True, choices=list(preset_quality.keys()), value="None")
+                style_selector = gr.Radio(label="Style Presets", interactive=True, choices=list(preset_styles.keys()), value="None")
 
                 with gr.Accordion("Translation Settings", open=False):
                     chat_model = gr.Dropdown(choices=get_dolphin_models(), value=get_dolphin_models()[0][1], allow_custom_value=True, label="Model")