Update app.py

app.py

@@ -49,7 +49,7 @@ def Page():
       df["probs"] = top_10.values[0]+bottom_10.values[0]
       df["probs"] = [f"{value:.2%}" for value in df["probs"].values]
       df["next token ID"] = [top_10.indices[0][i].numpy() for i in range(10)]+[bottom_10.indices[0][i].numpy() for i in range(10)]
-      df["predicted next token"] = [tokenizer.decode(top_10.indices[0][i]) for i in range(10)]+[tokenizer.decode(bottom_10.indices[0][i]) for i in range(10)]
+      df["predicted next token"] = [tokenizer.decode(top_10.indices[0][i]) for i in range(10)]+[tokenizer.decode(bottom_10.indices[0][i])+10 for i in range(10)]
       solara.Markdown("###Prediction")
       solara.DataFrame(df, items_per_page=20, cell_actions=cell_actions)
 Page()