schnitsum 0.1.1

No project description provided

Schnitsum: Easy to use neural network based summarization models

This package enables to generate summaries of you documents of interests.

Currently, we support following models,

• BART (large) fine-tuned on computer science papers (ref. SciTLDR).
  • Model name: sobamchan/bart-large-scitldr

we are planning to expand coverage soon to other domains, languages, models soon.

Installation

pip install schnitsum  # or poetry add schnitsum

This will let you generate summaries with CPUs only, if you want to utilize your GPUs, please follow the instruction by PyTorch, here.

Usage

from schnitsum import SchnitSum
model = SchnitSum("sobamchan/bart-large-scitldr")

docs = [
    "Document you want to summarize."
]

summaries = model(docs)
print(summaries)