Update app.py

app.py

@@ -159,11 +159,11 @@ def generate_markdown_table(scores, sequence_prob, top_k=4, chosen_tokens=None):
         </tr>"""
     for token_idx in np.array(np.argsort(scores)[-top_k:])[::-1]:
         token = tokenizer.decode([token_idx])
-        style = ""
+        item_class = ""
         if chosen_tokens and token in chosen_tokens:
-            style = "background-color:red"
+            item_class = "chosen"
         markdown_table += f"""
-        <tr style={style}>
+        <tr class={item_class}>
             <td>{token}</td>
             <td>{scores[token_idx]:.4f}</td>
             <td>{scores[token_idx] + sequence_prob:.4f}</td>