Downloads · 30 days
8
1% of all-time downloads
ehdwns1516/bart_finetuned_xsum
bart_finetuned_xsum is a machine learning model from ehdwns1516. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
This model has been trained as a xsum dataset. Input text what you want to summarize.
Downloads · 30 days
8
1% of all-time downloads
All-time downloads
1.5K
Public
Repo size
3.3 GB
Likes
0
Public
Click a slice to open those files.
.bin1.6 GB · 100%
From the Hugging Face model README
review generator DEMO: Ainize DEMO
review generator API: Ainize API
Language model: facebook/bart-large
Language: English
Training data: xsum dataset
Code: See Ainize Workspace
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("ehdwns1516/bart_finetuned_xsum")
model = AutoModelForSeq2SeqLM.from_pretrained("ehdwns1516/bart_finetuned_xsum")
summarizer = pipeline(
"summarization",
model="ehdwns1516/bart_finetuned_xsum",
tokenizer=tokenizer
)
context = "your context"
result = dict()
result[0] = summarizer(context)[0]