yasserrmd commited on
Commit
f15357c
·
verified ·
1 Parent(s): cabf737

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,7 +47,7 @@ def generate_image(occasion, theme, colors, randomize_seed=True, seed=0):
47
  num_inference_steps=50
48
  ).images[0]
49
 
50
- return image, seed
51
 
52
  css = """
53
  #col-container {
@@ -90,7 +90,7 @@ with gr.Blocks(css=css) as demo:
90
  triggers=[run_button.click],
91
  fn=generate_image,
92
  inputs=[occasion, theme, colors],
93
- outputs=[result],
94
  )
95
 
96
  demo.launch()
 
47
  num_inference_steps=50
48
  ).images[0]
49
 
50
+ return image
51
 
52
  css = """
53
  #col-container {
 
90
  triggers=[run_button.click],
91
  fn=generate_image,
92
  inputs=[occasion, theme, colors],
93
+ outputs=[result], # Expect only the image output
94
  )
95
 
96
  demo.launch()