dmmiller612 / bert-extractive-summarizer

Easy to use extractive text summarization with BERT
MIT License
1.39k stars 305 forks source link

Does this use pretrained models? #21

Closed dzimmerman-nci closed 4 years ago

dzimmerman-nci commented 4 years ago

Assuming so but want to confirm

davidlenz commented 4 years ago

yes, you can check them out here: https://github.com/dmmiller612/bert-extractive-summarizer/blob/master/summarizer/BertParent.py#L13

dmmiller612 commented 4 years ago

Yep, as David said, this uses the transformers library by huggingface.