prithivMLmods commited on
Commit
521e7b5
·
verified ·
1 Parent(s): 15e5fd4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
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: #2980b9 !important;
245
  color: white !important;
246
  }
247
  .submit-btn:hover {
248
- background-color: #3498db !important;
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(