fix colors

app.py

@@ -83,7 +83,7 @@ def prompt_to_layout(user_prompt, fpath=None):
     for poly in polygons:
         geom.append(Polygon(np.array(poly, dtype=int)))
         
-    colors = [architext_colors2[housegan_labels[space]] for space in spaces]
+    colors = [architext_colors[housegan_labels[space]] for space in spaces]
     
     _, im = draw_polygons(geom, colors, fpath=fpath)
     
@@ -109,7 +109,7 @@ def mut_txt2layout(mut_output):
     for poly in polygons:
         geom.append(Polygon(np.array(poly, dtype=int)))
 
-    colors = [architext_colors2[housegan_labels[space]] for space in spaces]
+    colors = [architext_colors[housegan_labels[space]] for space in spaces]
     _, im = draw_polygons(geom, colors, fpath=None)
 
     legend = Image.open("legend3.png")