Update app.py
Browse files
app.py
CHANGED
|
@@ -72,13 +72,13 @@ def infer(
|
|
| 72 |
guidance_scale,
|
| 73 |
num_inference_steps,
|
| 74 |
lora_scale,
|
| 75 |
-
use_controlnet,
|
| 76 |
-
controlnet_image,
|
| 77 |
-
controlnet_strength,
|
| 78 |
-
controlnet_mode,
|
| 79 |
-
use_ip_adapter,
|
| 80 |
-
ip_adapter_image,
|
| 81 |
-
ip_adapter_scale,
|
| 82 |
progress=gr.Progress(track_tqdm=True),
|
| 83 |
):
|
| 84 |
if randomize_seed:
|
|
@@ -191,59 +191,59 @@ with gr.Blocks(css=css) as demo:
|
|
| 191 |
# outputs=lora_section
|
| 192 |
# )
|
| 193 |
|
| 194 |
-
with gr.Group():
|
| 195 |
-
|
| 196 |
-
|
| 197 |
-
|
| 198 |
-
|
| 199 |
-
|
| 200 |
-
|
| 201 |
-
|
| 202 |
-
|
| 203 |
-
|
| 204 |
-
|
| 205 |
-
|
| 206 |
-
|
| 207 |
-
|
| 208 |
-
|
| 209 |
-
|
| 210 |
-
|
| 211 |
-
|
| 212 |
-
|
| 213 |
-
|
| 214 |
-
|
| 215 |
-
|
| 216 |
-
|
| 217 |
-
# Показывать/скрывать секцию ControlNet в зависимости от чекбокса
|
| 218 |
-
use_controlnet.change(
|
| 219 |
-
|
| 220 |
-
|
| 221 |
-
|
| 222 |
-
)
|
| 223 |
|
| 224 |
-
with gr.Group():
|
| 225 |
-
|
| 226 |
-
|
| 227 |
-
|
| 228 |
-
|
| 229 |
-
|
| 230 |
-
|
| 231 |
-
|
| 232 |
-
|
| 233 |
-
|
| 234 |
-
|
| 235 |
-
|
| 236 |
-
|
| 237 |
-
|
| 238 |
-
|
| 239 |
-
|
| 240 |
-
|
| 241 |
-
# Показывать/скрывать секцию IP-adapter в зависимости от чекбокса
|
| 242 |
-
use_ip_adapter.change(
|
| 243 |
-
|
| 244 |
-
|
| 245 |
-
|
| 246 |
-
)
|
| 247 |
|
| 248 |
with gr.Accordion("Advanced Settings", open=False):
|
| 249 |
negative_prompt = gr.Text(
|
|
@@ -300,6 +300,31 @@ with gr.Blocks(css=css) as demo:
|
|
| 300 |
|
| 301 |
gr.Examples(examples=examples, inputs=[prompt])
|
| 302 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 303 |
gr.on(
|
| 304 |
triggers=[run_button.click, prompt.submit],
|
| 305 |
fn=infer,
|
|
@@ -314,17 +339,9 @@ with gr.Blocks(css=css) as demo:
|
|
| 314 |
guidance_scale,
|
| 315 |
num_inference_steps,
|
| 316 |
lora_scale,
|
| 317 |
-
use_controlnet,
|
| 318 |
-
controlnet_image,
|
| 319 |
-
controlnet_strength,
|
| 320 |
-
controlnet_mode,
|
| 321 |
-
use_ip_adapter,
|
| 322 |
-
ip_adapter_image,
|
| 323 |
-
ip_adapter_scale,
|
| 324 |
],
|
| 325 |
outputs=[result, seed],
|
| 326 |
)
|
| 327 |
|
| 328 |
-
|
| 329 |
if __name__ == "__main__":
|
| 330 |
demo.launch()
|
|
|
|
| 72 |
guidance_scale,
|
| 73 |
num_inference_steps,
|
| 74 |
lora_scale,
|
| 75 |
+
use_controlnet=False,
|
| 76 |
+
controlnet_image=None,
|
| 77 |
+
controlnet_strength=None,
|
| 78 |
+
controlnet_mode=None,
|
| 79 |
+
use_ip_adapter=False,
|
| 80 |
+
ip_adapter_image=None,
|
| 81 |
+
ip_adapter_scale=None,
|
| 82 |
progress=gr.Progress(track_tqdm=True),
|
| 83 |
):
|
| 84 |
if randomize_seed:
|
|
|
|
| 191 |
# outputs=lora_section
|
| 192 |
# )
|
| 193 |
|
| 194 |
+
# with gr.Group():
|
| 195 |
+
# title = gr.Markdown(" ### ControlNet section")
|
| 196 |
+
# with gr.Column():
|
| 197 |
+
# use_controlnet = gr.Checkbox(label="Use ControlNet", value=False)
|
| 198 |
+
|
| 199 |
+
# # Секция ControlNet (изначально скрыта)
|
| 200 |
+
# with gr.Column(visible=False) as controlnet_section:
|
| 201 |
+
# controlnet_strength = gr.Slider(
|
| 202 |
+
# minimum=0.1, maximum=1.0, value=0.8, step=0.1,
|
| 203 |
+
# label="ControlNet Strength",
|
| 204 |
+
# interactive=True
|
| 205 |
+
# )
|
| 206 |
+
# controlnet_mode = gr.Dropdown(
|
| 207 |
+
# list(control_net_modes.keys()),
|
| 208 |
+
# value=next(iter(control_net_modes.keys())),
|
| 209 |
+
# label="ControlNet mode",
|
| 210 |
+
# interactive=True
|
| 211 |
+
# )
|
| 212 |
+
# controlnet_image = gr.Image(
|
| 213 |
+
# label="ControlNet image",
|
| 214 |
+
# type="pil"
|
| 215 |
+
# )
|
| 216 |
+
|
| 217 |
+
# # Показывать/скрывать секцию ControlNet в зависимости от чекбокса
|
| 218 |
+
# use_controlnet.change(
|
| 219 |
+
# fn=lambda x: gr.update(visible=x),
|
| 220 |
+
# inputs=use_controlnet,
|
| 221 |
+
# outputs=controlnet_section
|
| 222 |
+
# )
|
| 223 |
|
| 224 |
+
# with gr.Group():
|
| 225 |
+
# title = gr.Markdown(" ### IP-adapter section")
|
| 226 |
+
# with gr.Column():
|
| 227 |
+
# use_ip_adapter = gr.Checkbox(label="Use IP-adapter", value=False)
|
| 228 |
+
|
| 229 |
+
# # Секция IP-adapter (изначально скрыта)
|
| 230 |
+
# with gr.Column(visible=False) as ip_adapter_section:
|
| 231 |
+
# ip_adapter_scale = gr.Slider(
|
| 232 |
+
# minimum=0.1, maximum=1.0, value=0.5, step=0.1,
|
| 233 |
+
# label="IP-adapter Scale",
|
| 234 |
+
# interactive=True
|
| 235 |
+
# )
|
| 236 |
+
# ip_adapter_image = gr.Image(
|
| 237 |
+
# label="IP-adapter image",
|
| 238 |
+
# type="pil"
|
| 239 |
+
# )
|
| 240 |
+
|
| 241 |
+
# # Показывать/скрывать секцию IP-adapter в зависимости от чекбокса
|
| 242 |
+
# use_ip_adapter.change(
|
| 243 |
+
# fn=lambda x: gr.update(visible=x),
|
| 244 |
+
# inputs=use_ip_adapter,
|
| 245 |
+
# outputs=ip_adapter_section
|
| 246 |
+
# )
|
| 247 |
|
| 248 |
with gr.Accordion("Advanced Settings", open=False):
|
| 249 |
negative_prompt = gr.Text(
|
|
|
|
| 300 |
|
| 301 |
gr.Examples(examples=examples, inputs=[prompt])
|
| 302 |
|
| 303 |
+
# gr.on(
|
| 304 |
+
# triggers=[run_button.click, prompt.submit],
|
| 305 |
+
# fn=infer,
|
| 306 |
+
# inputs=[
|
| 307 |
+
# model_id,
|
| 308 |
+
# prompt,
|
| 309 |
+
# negative_prompt,
|
| 310 |
+
# seed,
|
| 311 |
+
# randomize_seed,
|
| 312 |
+
# width,
|
| 313 |
+
# height,
|
| 314 |
+
# guidance_scale,
|
| 315 |
+
# num_inference_steps,
|
| 316 |
+
# lora_scale,
|
| 317 |
+
# use_controlnet,
|
| 318 |
+
# controlnet_image,
|
| 319 |
+
# controlnet_strength,
|
| 320 |
+
# controlnet_mode,
|
| 321 |
+
# use_ip_adapter,
|
| 322 |
+
# ip_adapter_image,
|
| 323 |
+
# ip_adapter_scale,
|
| 324 |
+
# ],
|
| 325 |
+
# outputs=[result, seed],
|
| 326 |
+
# )
|
| 327 |
+
|
| 328 |
gr.on(
|
| 329 |
triggers=[run_button.click, prompt.submit],
|
| 330 |
fn=infer,
|
|
|
|
| 339 |
guidance_scale,
|
| 340 |
num_inference_steps,
|
| 341 |
lora_scale,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 342 |
],
|
| 343 |
outputs=[result, seed],
|
| 344 |
)
|
| 345 |
|
|
|
|
| 346 |
if __name__ == "__main__":
|
| 347 |
demo.launch()
|