Try to make labels

app.py

@@ -121,7 +121,10 @@ def prompt_to_layout(user_prompt, intensity, fpath=None):
     
     _, im = draw_polygons(geom, colors, fpath=fpath)
     
-    return im
+    html = '<img class="labels" src="labels.png" />'
+    
+    return im, html
+     
     
 # Gradio App
 
@@ -217,7 +220,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=["html", generated],
                       css=custom_css,
                       theme="default",
                       allow_flagging=False,