Update app.py

app.py

@@ -26,7 +26,7 @@ def get_next_token_probs(text):
     next_token_probs = torch.softmax(next_token_logits, dim=0)
     
     # Get top-5 tokens and their probabilities
-    topk_probs, topk_indices = torch.topk(next_token_probs, 5)
+    topk_probs, topk_indices = torch.topk(next_token_probs, 20)
     topk_tokens = [tokenizer.decode([idx]) for idx in topk_indices]
     
     # Format the results as strings