diff options
Diffstat (limited to 'modules/ui.py')
| -rw-r--r-- | modules/ui.py | 7 |
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,
|