UI changes for focal point image cropping

1 files changed, 3 insertions, 1 deletions

diff --git a/modules/ui.py b/modules/ui.py
index 1ff7eb4f..b6be713b 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1234,6 +1234,7 @@ def create_ui(wrap_gradio_gpu_call):
                     with gr.Row():

                         process_flip = gr.Checkbox(label='Create flipped copies')

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

+                        process_entropy_focus = gr.Checkbox(label='Create auto focal point crop')

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

                         process_caption_deepbooru = gr.Checkbox(label='Use deepbooru for caption', visible=True if cmd_opts.deepdanbooru else False)

 

@@ -1318,7 +1319,8 @@ def create_ui(wrap_gradio_gpu_call):
                 process_flip,

                 process_split,

                 process_caption,

-                process_caption_deepbooru

+                process_caption_deepbooru,

+                process_entropy_focus

             ],

             outputs=[

                 ti_output,