Update app.py

app.py

@@ -144,7 +144,7 @@ def generate_markdown_table(scores, top_k=4, chosen_tokens=None):
     <table>
         <tr>
             <th><b>Token</b></th>
-            <th><b>Probability</b></th>
+            <th><b>Score</b></th>
         </tr>"""
     for token_idx in np.argsort(scores)[-top_k:]:
         token = tokenizer.decode([token_idx])
@@ -154,7 +154,7 @@ def generate_markdown_table(scores, top_k=4, chosen_tokens=None):
         markdown_table += f"""
         <tr style={style}>
             <td>{token}</td>
-            <td>{scores[token_idx]}</td>
+            <td>{scores[token_idx]:.4f}</td>
         </tr>"""
     markdown_table += """
     </table>"""