Update app.py

app.py

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