Thanks lewtun!
Can you recommend any other type of model for extractive summarization, especially if I want to train it further?
Also, how to make sure that the summaries aren’t cut short in between? I am getting sentences like “your sling. Cut a piece of fabric …”. “Your sling” is a portion of the text that was cut in middle. How to avoid this?
I tried two ways :- one is
i)changing max_length in  model.generate(input_ids, attention_mask=attention_mask,max_length=300)
2) varying encoder_max_length in ‘batch_tokenize’
which is the correct method? If none, can you suggest?