Update app.py

app.py

@@ -131,7 +131,7 @@ def prompt_to_layout(user_prompt, intensity, fpath=None):
     imgs_comb = np.vstack([im, legend])
     imgs_comb = Image.fromarray(imgs_comb)
     
-    return imgs_comb
+    return imgs_comb, layout 
      
     
 # Gradio App
@@ -259,7 +259,7 @@ textbox = gr.inputs.Textbox(placeholder='An apartment with two bedrooms and one
 generated = gr.outputs.Image(label='Generated Layout')
 
 iface = gr.Interface(fn=prompt_to_layout, inputs=[textbox, creative_slider], 
-                     outputs=[generated],
+                     outputs=[generated, layout],
                      css=custom_css,
                      theme="default",
                      allow_flagging='never',