feat: automatically describe image on uov image upload (#1938)
* feat: automatically describe image on uov image upload if prompt is empty * feat: add argument to disable automatic uov image description * feat: rename argument, disable by default this prevents computers with low hardware specifications from being unnecessary blocked
This commit is contained in:
parent
1eb58fa366
commit
33fa175bd4
|
|
@ -31,6 +31,9 @@ args_parser.parser.add_argument("--disable-metadata", action='store_true',
|
||||||
args_parser.parser.add_argument("--disable-preset-download", action='store_true',
|
args_parser.parser.add_argument("--disable-preset-download", action='store_true',
|
||||||
help="Disables downloading models for presets", default=False)
|
help="Disables downloading models for presets", default=False)
|
||||||
|
|
||||||
|
args_parser.parser.add_argument("--enable-describe-uov-image", action='store_true',
|
||||||
|
help="Disables automatic description of uov images when prompt is empty", default=False)
|
||||||
|
|
||||||
args_parser.parser.add_argument("--always-download-new-model", action='store_true',
|
args_parser.parser.add_argument("--always-download-new-model", action='store_true',
|
||||||
help="Always download newer models ", default=False)
|
help="Always download newer models ", default=False)
|
||||||
|
|
||||||
|
|
|
||||||
9
webui.py
9
webui.py
|
|
@ -717,6 +717,15 @@ with shared.gradio_root:
|
||||||
desc_btn.click(trigger_describe, inputs=[desc_method, desc_input_image],
|
desc_btn.click(trigger_describe, inputs=[desc_method, desc_input_image],
|
||||||
outputs=[prompt, style_selections], show_progress=True, queue=True)
|
outputs=[prompt, style_selections], show_progress=True, queue=True)
|
||||||
|
|
||||||
|
if args_manager.args.enable_describe_uov_image:
|
||||||
|
def trigger_uov_describe(mode, img, prompt):
|
||||||
|
# keep prompt if not empty
|
||||||
|
if prompt == '':
|
||||||
|
return trigger_describe(mode, img)
|
||||||
|
return gr.update(), gr.update()
|
||||||
|
|
||||||
|
uov_input_image.upload(trigger_uov_describe, inputs=[desc_method, uov_input_image, prompt],
|
||||||
|
outputs=[prompt, style_selections], show_progress=True, queue=True)
|
||||||
|
|
||||||
def dump_default_english_config():
|
def dump_default_english_config():
|
||||||
from modules.localization import dump_english_config
|
from modules.localization import dump_english_config
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue