Update README.md

README.md

@@ -69,7 +69,6 @@ Given these potential pitfalls, and others not explicitly mentioned, it's essent
 ### License
 The model is licensed under [Research License](https://huggingface.co/microsoft/phi-1/resolve/main/Research%20License.docx).
 
-
 ### Sample Code
 ```python
 import torch
@@ -83,7 +82,7 @@ inputs = tokenizer('''def print_prime(n):
    Print all primes between 1 and n
    """''', return_tensors="pt", return_attention_mask=False)
 
-outputs = model.generate(**inputs, max_length=500, eos_token_id=50256, pad_token_id=50256)
+outputs = model.generate(**inputs, max_length=200)
 text = tokenizer.batch_decode(outputs)[0]
 print(text)
 ```