Edit model card

The finetuned BART-Large based model is used to expand abstract sentences. Use the following demo code to generate diverse expansions from abstract inputs.

Code

from transformers import pipeline, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained('utkarsh4430/ABEX-abstract-expand')
model_pipeline = pipeline("text2text-generation", model='utkarsh4430/ABEX-abstract-expand', tokenizer=tokenizer)

input_text = 'A chance to meet WWE stars and support a good cause.'

for f in range(5):
    generated_text = model_pipeline(input_text, num_beams=1, top_k=100, do_sample=True, max_length=350, num_return_sequences=1)
    print(f'Aug {f}: ', generated_text[0]['generated_text'])

Example output:

Aug 0:  WWE stars to visit Detroit on January 20th for the second time in three years, with appearances at The Battle at the Fox Sports 1 World Headquarters, and proceeds going to a charity of your choice.
Aug 1:  A one-on-one experience at the WWE Creative Conference in New Jersey was provided, with an opportunity for the audience to meet WWE superstars and support a good cause.
Aug 2:  Sindrun welcomes WWE star Chris Jericho and hosts an event for attendees to meet WWE stars and support a local cause.
Aug 3:  Find out if you can meet WWE stars, including the Rock and Shake, at a benefit luncheon.
Aug 4:  The WWE Talent Showcase 2019 will feature exciting moments inside the WWE Studios, including the first one in over a decade, and features a chance to hug current and former stars and receive a check from a corporate sponsor.

Citation instructions:

@inproceedings{
ghosh2024abex,
title={{ABEX}: Data Augmentation for Low-Resource {NLU} via Expanding Abstract Descriptions},
author={Sreyan Ghosh and Utkarsh Tyagi and Sonal Kumar and Chandra Kiran Reddy Evuru and and Ramaneswaran S and S Sakshi and Dinesh Manocha},
booktitle={The 62nd Annual Meeting of the Association for Computational Linguistics},
year={2024},
}
Downloads last month
8
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.