Update README.md

README.md

@@ -40,7 +40,8 @@ from transformers import pipeline
 #Feel free to play around with the generation parameters.
 #Reduce the beam width for faster inference
 #Note that the maximum length for the generated titles is 64
-gen_kwargs = {"length_penalty": 0.8, "num_beams":8, "max_length": 48} 
+gen_kwargs = {"length_penalty": 0.6, "num_beams":4, "num_return_sequences": 4,"num_beam_groups":4,"diversity_penalty":2.0}
+
 pipe = pipeline("summarization", model="RajSang/pegasus-sports-titles")
 
 #Change the article according to your wish