AP123 PeterL1n commited on
Commit
f2aa102
1 Parent(s): 3d2c0fd

Update app.py (#4)

Browse files

- Update app.py (d484f91b36fc1f2c9fc069a3fa9bd2f12fd28c34)


Co-authored-by: Peter Lin <[email protected]>

Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -75,19 +75,19 @@ def generate_image(prompt, ckpt):
75
 
76
  # Gradio Interface
77
  description = """
78
- This demo utilizes the SDXL-Lightning model by ByteDance, which is a fast text-to-image generative model capable of producing high-quality images in 4 steps.
79
  As a community effort, this demo was put together by AngryPenguin. Link to model: https://huggingface.co/ByteDance/SDXL-Lightning
80
  """
81
 
82
  with gr.Blocks(css="style.css") as demo:
83
- gr.HTML("<h1><center>Text-to-Image with SDXL Lightning ⚡</center></h1>")
84
  gr.Markdown(description)
85
  with gr.Group():
86
  with gr.Row():
87
  prompt = gr.Textbox(label='Enter you image prompt:', scale=8)
88
- ckpt = gr.Dropdown(label='Select Inference Steps',choices=['1-Step', '2-Step', '4-Step', '8-Step'], value='4-Step', interactive=True)
89
  submit = gr.Button(scale=1, variant='primary')
90
- img = gr.Image(label='SDXL-Lightening Generate Image')
91
 
92
  prompt.submit(fn=generate_image,
93
  inputs=[prompt, ckpt],
 
75
 
76
  # Gradio Interface
77
  description = """
78
+ This demo utilizes the SDXL-Lightning model by ByteDance, which is a lightning-fast text-to-image generative model capable of producing high-quality images in 4 steps.
79
  As a community effort, this demo was put together by AngryPenguin. Link to model: https://huggingface.co/ByteDance/SDXL-Lightning
80
  """
81
 
82
  with gr.Blocks(css="style.css") as demo:
83
+ gr.HTML("<h1><center>Text-to-Image with SDXL-Lightning ⚡</center></h1>")
84
  gr.Markdown(description)
85
  with gr.Group():
86
  with gr.Row():
87
  prompt = gr.Textbox(label='Enter you image prompt:', scale=8)
88
+ ckpt = gr.Dropdown(label='Select inference steps',choices=['1-Step', '2-Step', '4-Step', '8-Step'], value='4-Step', interactive=True)
89
  submit = gr.Button(scale=1, variant='primary')
90
+ img = gr.Image(label='SDXL-Lightning Generated Image')
91
 
92
  prompt.submit(fn=generate_image,
93
  inputs=[prompt, ckpt],