Update model.py

model.py

@@ -147,7 +147,7 @@ model.load_state_dict(torch.load("model_weights.pth", map_location=device))
 model.eval()
 
 # Prompt
-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)
+context = torch.tensor([enc.encode("In a faraway land, ")], dtype=torch.long, device=device)
 
 # Test generation with a higher number of tokens and adjusted temperature
 max_new_tokens = 150  # Increase the token limit for a longer generation