swap layer

app.py

@@ -41,8 +41,8 @@ def gpu_decorator(func):
 
 
 # Constants
-layer_names = ['attn', 'ff']
-colors_rgb = [(255, 103, 35), (0, 210, 106)]  # orange, green
+layer_names = ['ff', 'attn']
+colors_rgb = [(0, 210, 106), (255, 103, 35)]  # green, orange
 cell_size = 20
 spacing = 2
 n_layers = 2