Merge pull request #2143 from JC-Array/deepdanbooru_pre_process

deepbooru tags for textual inversion preproccessing

1 files changed, 6 insertions, 1 deletions

diff --git a/modules/ui.py b/modules/ui.py
index 33a49d3b..00bf09ae 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -324,7 +324,7 @@ def interrogate(image):
 

 

 def interrogate_deepbooru(image):

-    prompt = get_deepbooru_tags(image, opts.interrogate_deepbooru_score_threshold)

+    prompt = get_deepbooru_tags(image)

     return gr_show(True) if prompt is None else prompt

 

 

@@ -1065,6 +1065,10 @@ def create_ui(wrap_gradio_gpu_call):
                         process_flip = gr.Checkbox(label='Create flipped copies')

                         process_split = gr.Checkbox(label='Split oversized images into two')

                         process_caption = gr.Checkbox(label='Use BLIP caption as filename')

+                        if cmd_opts.deepdanbooru:

+                            process_caption_deepbooru = gr.Checkbox(label='Use deepbooru caption as filename')

+                        else:

+                            process_caption_deepbooru = gr.Checkbox(label='Use deepbooru caption as filename', visible=False)

 

                     with gr.Row():

                         with gr.Column(scale=3):

@@ -1142,6 +1146,7 @@ def create_ui(wrap_gradio_gpu_call):
                 process_flip,

                 process_split,

                 process_caption,

+                process_caption_deepbooru

             ],

             outputs=[

                 ti_output,