Kabatubare commited on
Commit
9f7adf4
·
verified ·
1 Parent(s): 5e07b28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -8
app.py CHANGED
@@ -72,22 +72,20 @@ def generate_image(prompt, ckpt):
72
  return results.images[0]
73
 
74
 
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 your prompt (English)', 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],
@@ -98,4 +96,4 @@ with gr.Blocks(css="style.css") as demo:
98
  outputs=img,
99
  )
100
 
101
- demo.queue().launch()
 
72
  return results.images[0]
73
 
74
 
 
75
  # Gradio Interface
76
  description = """
77
+ Welcome aboard the Starship SDXL Enterprise! Our mission: to explore strange new AI generations, to seek out new visual frontiers and computational boundaries, to boldly generate images like no one has seen before. Utilizing the cutting-edge SDXL-Lightning model, we're at the forefront of text-to-image technology, ready to transform your imaginative prompts into visual spectacles. Whether you're navigating the uncharted territories of outer space or delving into the realms of fantasy, your adventure begins now. Model powered by the pioneering intellects at ByteDance. Journey safely through the stars!
 
78
  """
79
 
80
  with gr.Blocks(css="style.css") as demo:
81
+ gr.HTML("<h1><center>🌌 Starfleet Command: Text-to-Image Warp Drive - SDXL-Lightning ⚡</center></h1>")
82
  gr.Markdown(description)
83
  with gr.Group():
84
  with gr.Row():
85
+ prompt = gr.Textbox(label='Your cosmic prompt (English)', scale=8)
86
+ ckpt = gr.Dropdown(label='Warp Factor (Inference Speed)', choices=['Warp 1', 'Warp 2', 'Warp 4', 'Warp 8'], value='Warp 4', interactive=True)
87
  submit = gr.Button(scale=1, variant='primary')
88
+ img = gr.Image(label='The Universe, As You Envision It')
89
 
90
  prompt.submit(fn=generate_image,
91
  inputs=[prompt, ckpt],
 
96
  outputs=img,
97
  )
98
 
99
+ demo.queue().launch()