Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -49,6 +49,13 @@ pipelines = {
|
|
| 49 |
controlnet=model,
|
| 50 |
variant="fp16",
|
| 51 |
).to("cuda"),
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
}
|
| 53 |
|
| 54 |
for pipe in pipelines.values():
|
|
@@ -241,11 +248,11 @@ h1 {
|
|
| 241 |
display: block;
|
| 242 |
}
|
| 243 |
.submit-btn {
|
| 244 |
-
background-color: #
|
| 245 |
color: white !important;
|
| 246 |
}
|
| 247 |
.submit-btn:hover {
|
| 248 |
-
background-color: #
|
| 249 |
"""
|
| 250 |
|
| 251 |
title = """<h1 align="center">Diffusers Image Outpaint Lightning</h1>
|
|
@@ -266,7 +273,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
|
|
| 266 |
with gr.Column(scale=2):
|
| 267 |
prompt_input = gr.Textbox(label="Prompt (Optional)")
|
| 268 |
with gr.Column(scale=1):
|
| 269 |
-
run_button = gr.Button("Generate Image / Diffusers Outpaint Image Lightning / Lightning v4, v5", elem_classes="submit-btn")
|
| 270 |
|
| 271 |
with gr.Row():
|
| 272 |
model_selector = gr.Dropdown(
|
|
|
|
| 49 |
controlnet=model,
|
| 50 |
variant="fp16",
|
| 51 |
).to("cuda"),
|
| 52 |
+
"RealVisXL_V5.0": StableDiffusionXLFillPipeline.from_pretrained(
|
| 53 |
+
"SG161222/RealVisXL_V5.0",
|
| 54 |
+
torch_dtype=torch.float16,
|
| 55 |
+
vae=vae,
|
| 56 |
+
controlnet=model,
|
| 57 |
+
variant="fp16",
|
| 58 |
+
).to("cuda"),
|
| 59 |
}
|
| 60 |
|
| 61 |
for pipe in pipelines.values():
|
|
|
|
| 248 |
display: block;
|
| 249 |
}
|
| 250 |
.submit-btn {
|
| 251 |
+
background-color: #D8BFD8 !important;
|
| 252 |
color: white !important;
|
| 253 |
}
|
| 254 |
.submit-btn:hover {
|
| 255 |
+
background-color: #DDA0DD !important;
|
| 256 |
"""
|
| 257 |
|
| 258 |
title = """<h1 align="center">Diffusers Image Outpaint Lightning</h1>
|
|
|
|
| 273 |
with gr.Column(scale=2):
|
| 274 |
prompt_input = gr.Textbox(label="Prompt (Optional)")
|
| 275 |
with gr.Column(scale=1):
|
| 276 |
+
run_button = gr.Button("Re-Generate Image / Diffusers Outpaint Image Lightning / Lightning v4, v5", elem_classes="submit-btn")
|
| 277 |
|
| 278 |
with gr.Row():
|
| 279 |
model_selector = gr.Dropdown(
|