Update model.py

model.py

@@ -150,7 +150,7 @@ model.eval()
 context = torch.tensor([enc.encode("Once upon a time there was a knight called Bob and he rode into his greatest battle yet")], dtype=torch.long, device=device)
 
 # Test generation with a higher number of tokens and adjusted temperature
-max_new_tokens = 200  # Increase the token limit for a longer generation
+max_new_tokens = 150  # Increase the token limit for a longer generation
 temperature = 0.6  # More focused, less random
 generated_text_idx = model.generate(context, max_new_tokens)
 generated_text = enc.decode(generated_text_idx[0].tolist())