Update README.md

README.md

@@ -43,8 +43,8 @@ input_ids = tokenizer.encode(
 
 # Generate
 output = model.generate(
-    input_ids, max_length=150, num_return_sequences=1, 
-    no_repeat_ngram_size=2, early_stopping=True
+    input_ids, max_length=60, num_return_sequences=1, 
+    early_stopping=True, pad_token_id=tokenizer.eos_token_id,
 )
 
 # Decode output