Update app.py

app.py

@@ -68,7 +68,7 @@ def analyze_next_token(input_text, temperature, top_p, top_k):
         probabilities = torch.nn.functional.softmax(last_token_logits, dim=-1)
         top_k = 10
         top_probs, top_indices = torch.topk(probabilities, top_k)
-        top_words = [tokenizer.decode([idx.item()]).strip() for idx in top_indices]
+        top_words = [tokenizer.decode(idx.item()).strip() for idx in top_indices]
         prob_data = {word: prob.item() for word, prob in zip(top_words, top_probs)}
         prob_plot = plot_probabilities(prob_data)