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(
|