Skip to content

Instantly share code, notes, and snippets.

@manmohan24nov

manmohan24nov/reddit_xlnet_summarization.py

Created November 10, 2020 19:04
Show Gist options
  • Save manmohan24nov/df620695d3720a1b88f1fc6a4ac905ce to your computer and use it in GitHub Desktop.
Save manmohan24nov/df620695d3720a1b88f1fc6a4ac905ce to your computer and use it in GitHub Desktop.
Download ZIP
Raw
reddit_xlnet_summarization.py
>>> from summarizer import TransformerSummarizer
>>> xlnet_model = TransformerSummarizer(transformer_type="XLNet",transformer_model_key="xlnet-base-cased")
>>> text = " ".join(comment_section)
>>> summerize = ''.join(xlnet_model(text, min_length=60, max_length=120))
>>> summerize
'You can also get one for your 60th wedding anniversary and possibly some others. Only those who apply get one.
you get one at 100 and one every year from 105.except there is no queen of england she died over 100 years ago.'
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment