Fix Online Softmax charts: increase height (350px), margins, and y-axis range for labels

src/visualizer.py

@@ -281,13 +281,14 @@ def create_online_softmax_state(
     )
     
     fig.update_layout(
-        height=280,
-        margin=dict(l=40, r=40, t=60, b=40),
+        height=350,
+        margin=dict(l=40, r=40, t=80, b=40),
         showlegend=False,
     )
     
-    fig.update_yaxes(range=[0, 12], row=1, col=1)
-    fig.update_yaxes(range=[0, 15], row=1, col=2)
+    # Increase y-axis range to make room for text labels above bars
+    fig.update_yaxes(range=[0, 14], row=1, col=1)
+    fig.update_yaxes(range=[0, 18], row=1, col=2)
     
     # Generate explanation text
     d = current_data