Update app.py

app.py

@@ -83,7 +83,7 @@ def prompt_to_layout(user_prompt, top_p, top_k, 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(r"F:\Personal\Startup\ArchiText\Media\legend.png")