Fix experiments section visibility and improve ablation logic - Fix experiments section callback to show properly after analysis - Change mean ablation to use mean across ALL layers and tokens (not just selected layer) - Update UI description to clarify meaningless mean activation - Remove prevent_initial_call to ensure proper initialization
Add Experiments panel with mean ablation feature - Remove Divergent Layers summary from comparison panel (keep badges) - Add collapsible Experiments section after analysis - Implement Ablation Study with layer selection buttons - Add mean ablation function to replace layer activations with mean - Run ablation on both prompts and update results - Gray out ablated layer in accordion with visual indicator - Add CSS styling for experiments panel and ablated layers
Refactor: Move attention head categorization to on-demand per-layer display - Removed automatic Attention Head Categorization section from main panel - Created categorize_single_layer_heads() for per-layer categorization - Updated run_analysis to skip automatic head categorization - Enhanced node click callback to show layer-specific categorization - Analysis Results section now displays both BertViz and categorized heads on node click