multimodalart HF staff commited on
Commit
56acc31
1 Parent(s): d465e66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,7 +14,7 @@ import spaces
14
  os.environ['TOKENIZERS_PARALLELISM'] = 'false'
15
 
16
  DESCRIPTION = "# Stable Cascade"
17
- DESCRIPTION += "\n<p style=\"text-align: center\"><a href='https://huggingface.co/stabilityai/stable-cascade' target='_blank'>Stable Casaade</a> is a new fast and efficient high resolution text-to-image architecture and model built on the Würstchen architecture</p>"
18
  if not torch.cuda.is_available():
19
  DESCRIPTION += "\n<p>Running on CPU 🥶</p>"
20
 
@@ -133,7 +133,7 @@ def generate(
133
  # },
134
  # )
135
 
136
- yield decoder_output
137
 
138
 
139
  examples = [
@@ -158,7 +158,7 @@ with gr.Blocks() as demo:
158
  container=False,
159
  )
160
  run_button = gr.Button("Run", scale=0)
161
- result = gr.Gallery(label="Result", show_label=False)
162
  with gr.Accordion("Advanced options", open=False):
163
  negative_prompt = gr.Text(
164
  label="Negative prompt",
 
14
  os.environ['TOKENIZERS_PARALLELISM'] = 'false'
15
 
16
  DESCRIPTION = "# Stable Cascade"
17
+ DESCRIPTION += "\n<p style=\"text-align: center\"><a href='https://huggingface.co/stabilityai/stable-cascade' target='_blank'>Stable Casaade</a> is a high resolution text-to-image model built on the Würstchen architecture, efficient by using a smaller latent space</p>"
18
  if not torch.cuda.is_available():
19
  DESCRIPTION += "\n<p>Running on CPU 🥶</p>"
20
 
 
133
  # },
134
  # )
135
 
136
+ yield decoder_output[0]
137
 
138
 
139
  examples = [
 
158
  container=False,
159
  )
160
  run_button = gr.Button("Run", scale=0)
161
+ result = gr.Image(label="Result", show_label=False)
162
  with gr.Accordion("Advanced options", open=False):
163
  negative_prompt = gr.Text(
164
  label="Negative prompt",