pharmapsychotic/clip-interrogator-ext
1
0
Fork 0
mirror of https://github.com/pharmapsychotic/clip-interrogator-ext.git synced 2026-01-26 19:29:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add "Send to" buttons, thanks to @Dahvikiin

Browse Source
This commit is contained in:
pharmapsychotic
2023-09-09 20:35:46 -05:00
parent 661fbd95c4
commit 0f1a4591f8
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/clip_interrogator_ext.py
View File

@@ -10,6 +10,7 @@ from PIL import Image
import clip_interrogator import clip_interrogator
from clip_interrogator import Config, Interrogator from clip_interrogator import Config, Interrogator
import modules.generation_parameters_copypaste as parameters_copypaste
from modules import devices, lowvram, script_callbacks, shared from modules import devices, lowvram, script_callbacks, shared
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
@@ -17,7 +18,7 @@ from fastapi import FastAPI
from fastapi.exceptions import HTTPException from fastapi.exceptions import HTTPException
from io import BytesIO from io import BytesIO
__version__ = "0.2.0" __version__ = "0.2.1"
ci = None ci = None
low_vram = False low_vram = False
@@ -277,9 +278,14 @@ def prompt_tab():
with gr.Row(): with gr.Row():
button = gr.Button("Generate", variant='primary') button = gr.Button("Generate", variant='primary')
unload_button = gr.Button("Unload") unload_button = gr.Button("Unload")
with gr.Row():
buttons = parameters_copypaste.create_buttons(["txt2img", "img2img", "inpaint", "extras"])
button.click(image_to_prompt, inputs=[image, mode, clip_model], outputs=prompt) button.click(image_to_prompt, inputs=[image, mode, clip_model], outputs=prompt)
unload_button.click(unload) unload_button.click(unload)
for tabname, button in buttons.items():
parameters_copypaste.register_paste_params_button(parameters_copypaste.ParamBinding(paste_button=button, tabname=tabname, source_text_component=prompt, source_image_component=image,))
def add_tab(): def add_tab():
global low_vram global low_vram