prithivMLmods commited on
Commit
3f564f5
·
verified ·
1 Parent(s): 2a821e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -9
app.py CHANGED
@@ -300,7 +300,7 @@ else:
300
  use_safetensors=True,
301
  add_watermarker=False,
302
  ).to(device)
303
- print("Model Loaded !!")
304
 
305
  # A dictionary to easily choose the model based on selection.
306
  DEFAULT_MODEL = "RealVisXL_V5.0_Lightning"
@@ -401,8 +401,16 @@ examples = [
401
  ]
402
 
403
  css = '''
404
- .gradio-container{max-width: 888px !important}
405
- h1{text-align:center}
 
 
 
 
 
 
 
 
406
  .submit-btn {
407
  background: linear-gradient(90deg, #4B79A1 0%, #283E51 100%) !important;
408
  border: none !important;
@@ -425,12 +433,6 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
425
  )
426
  run_button = gr.Button("Generate as (1024 x 1024)🍺", scale=0, elem_classes="submit-btn")
427
 
428
- with gr.Row(visible=True):
429
- model_selection = gr.Dropdown(
430
- choices=MODEL_CHOICES,
431
- value=DEFAULT_MODEL,
432
- label="Model Selection",
433
- )
434
  with gr.Row(visible=True):
435
  grid_size_selection = gr.Dropdown(
436
  choices=["2x1", "1x2", "2x2", "2x3", "3x2", "1x1"],
@@ -464,6 +466,12 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
464
  value=DEFAULT_STYLE_NAME,
465
  label="Quality Style",
466
  )
 
 
 
 
 
 
467
  with gr.Accordion("Advanced options", open=False):
468
  use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True, visible=True)
469
  negative_prompt = gr.Text(
 
300
  use_safetensors=True,
301
  add_watermarker=False,
302
  ).to(device)
303
+ print("Running on CPU; models loaded in float32.")
304
 
305
  # A dictionary to easily choose the model based on selection.
306
  DEFAULT_MODEL = "RealVisXL_V5.0_Lightning"
 
401
  ]
402
 
403
  css = '''
404
+ .gradio-container {
405
+ max-width: 888px !important;
406
+ margin: 0 auto !important;
407
+ display: flex;
408
+ flex-direction: column;
409
+ align-items: center;
410
+ }
411
+ h1 {
412
+ text-align: center;
413
+ }
414
  .submit-btn {
415
  background: linear-gradient(90deg, #4B79A1 0%, #283E51 100%) !important;
416
  border: none !important;
 
433
  )
434
  run_button = gr.Button("Generate as (1024 x 1024)🍺", scale=0, elem_classes="submit-btn")
435
 
 
 
 
 
 
 
436
  with gr.Row(visible=True):
437
  grid_size_selection = gr.Dropdown(
438
  choices=["2x1", "1x2", "2x2", "2x3", "3x2", "1x1"],
 
466
  value=DEFAULT_STYLE_NAME,
467
  label="Quality Style",
468
  )
469
+ with gr.Row(visible=True):
470
+ model_selection = gr.Dropdown(
471
+ choices=MODEL_CHOICES,
472
+ value=DEFAULT_MODEL,
473
+ label="Model Selection",
474
+ )
475
  with gr.Accordion("Advanced options", open=False):
476
  use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True, visible=True)
477
  negative_prompt = gr.Text(