Update README.md

README.md

@@ -24,7 +24,7 @@ text = 'This is a perfect way to break the ice and start the conversation.'
 
 inputs = prompt.format(task, text)
 inputs = tokenizer(inputs, return_tensors="pt")
-output = model.generate(**inputs)
+output = model.generate(**inputs, num_beams=5, max_length=10)
 pred = tokenizer.decode(output[0].tolist(), skip_special_tokens=True, clean_up_tokenization_spaces=False)
 ```