Spaces:
Runtime error
Runtime error
Update app.py (#4)
Browse files- Update app.py (d484f91b36fc1f2c9fc069a3fa9bd2f12fd28c34)
Co-authored-by: Peter Lin <[email protected]>
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
|
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
|
89 |
submit = gr.Button(scale=1, variant='primary')
|
90 |
-
img = gr.Image(label='SDXL-
|
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],
|