Fix hover

frontend/index.html

@@ -117,7 +117,7 @@
       <section class="hover-visualization">
         <h2>Hover Visualization</h2>
         <p>
-          Hover over the tokens to see the attention weights for the selected head. Click on a token to see the attention weights for that token across all heads.
+          By hovering over each token, you can see which other token is important for that token. The larger the token, the more important it is for the token you are hovering over. The token with the maximal attention is colored in red.
         </p>
         <div id="tokenContainer"></div>
       </section>

frontend/script.js

@@ -180,7 +180,7 @@ function highlightAttention(index, attentionData, layer_idx, head_idx) {
   }
   
   const weights = row.slice(0, index);
-  const maxWeight = Math.max(...attentionData[layer_idx][head_idx]) || 1;
+  const maxWeight = Math.max(...weights) || 1;
   const baseFontSize = 32;
   const maxIncrease = 20;