Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-26 01:30:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Clean up send_pictures extension

Browse Source
This commit is contained in:
oobabooga 2023-02-25 02:13:24 -03:00
parent 01acb250c5
commit ad83b1b828
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
extensions/send_pictures/script.py
View File

@ -10,11 +10,10 @@ from modules.bot_picture import caption_image
params = { params = {
} }
# If 'state' is 'temporary' or 'permanent', will hijack the next # If 'state' is True, will hijack the next chat generation with
# chatbot wrapper call with a custom input text and optionally # custom input text
# custom output text
input_hijack = { input_hijack = {
'state': 'off', 'state': False,
'value': ["", ""] 'value': ["", ""]
} }
@ -57,4 +56,3 @@ def ui():
picture_select.upload(lambda picture, name1, name2: input_hijack.update({"state": True, "value": generate_chat_picture(picture, name1, name2)}), [picture_select, shared.gradio['name1'], shared.gradio['name2']], None) picture_select.upload(lambda picture, name1, name2: input_hijack.update({"state": True, "value": generate_chat_picture(picture, name1, name2)}), [picture_select, shared.gradio['name1'], shared.gradio['name2']], None)
picture_select.upload(eval(function_call), shared.input_params, shared.gradio['display'], show_progress=shared.args.no_stream) picture_select.upload(eval(function_call), shared.input_params, shared.gradio['display'], show_progress=shared.args.no_stream)
picture_select.upload(lambda : None, [], [picture_select], show_progress=False) picture_select.upload(lambda : None, [], [picture_select], show_progress=False)
#parser.add_argument('--picture', action='store_true', help='Adds an ability to send pictures in chat UI modes. Captions are generated by BLIP.')