Update app.py

app.py

@@ -41,7 +41,7 @@ def Page():
           font-size: 0.45em"> {token}</span>{tokenizer.decode([token])}</span>"""
       solara.Markdown(f'{spans2}')
       solara.Markdown(f'{spans1}')
-      outputs = model.generate(tokens, max_new_tokens=1, output_scores=True, return_dict_in_generate=True, pad_token_id=tokenizer.eos_token_id)
+      outputs = model.generate(tokens, max_new_tokens=2, output_scores=True, return_dict_in_generate=True, pad_token_id=tokenizer.eos_token_id)
       scores = F.softmax(outputs.scores[0], dim=-1)
       top_10 = torch.topk(scores, 100)
       df = pd.DataFrame()