THEODOROS commited on
Commit
97bfe1e
·
1 Parent(s): 1ee4a39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -131,7 +131,7 @@ def prompt_to_layout(user_prompt, intensity, fpath=None):
131
  imgs_comb = np.vstack([im, legend])
132
  imgs_comb = Image.fromarray(imgs_comb)
133
 
134
- return imgs_comb
135
 
136
 
137
  # Gradio App
@@ -259,7 +259,7 @@ textbox = gr.inputs.Textbox(placeholder='An apartment with two bedrooms and one
259
  generated = gr.outputs.Image(label='Generated Layout')
260
 
261
  iface = gr.Interface(fn=prompt_to_layout, inputs=[textbox, creative_slider],
262
- outputs=[generated],
263
  css=custom_css,
264
  theme="default",
265
  allow_flagging='never',
 
131
  imgs_comb = np.vstack([im, legend])
132
  imgs_comb = Image.fromarray(imgs_comb)
133
 
134
+ return imgs_comb, layout
135
 
136
 
137
  # Gradio App
 
259
  generated = gr.outputs.Image(label='Generated Layout')
260
 
261
  iface = gr.Interface(fn=prompt_to_layout, inputs=[textbox, creative_slider],
262
+ outputs=[generated, layout],
263
  css=custom_css,
264
  theme="default",
265
  allow_flagging='never',