Adding labels to image directly

app.py

@@ -123,7 +123,13 @@ def prompt_to_layout(user_prompt, intensity, fpath=None):
     
     html = '<img class="labels" src="images/labels.png" />'
     
-    return html, im
+    legend = Image.open("images/labels.png")
+    
+    im = np.array(im)
+    im[:40, :] = np.array(legend)
+    im = Image.fromarray(im)
+    
+    return im
      
     
 # Gradio App