Idea is to build a model which will take keywords as inputs and generate sentences as outputs.
Two Models have been built:
- Using T5-base size = 850 MB can be found here: https://huggingface.co/gagan3012/keytotext
- Using T5-small size = 230 MB can be found here: https://huggingface.co/gagan3012/keytotext-small
from transformers import AutoTokenizer, AutoModelWithLMHead tokenizer = AutoTokenizer.from_pretrained("gagan3012/keytotext-small") model = AutoModelWithLMHead.from_pretrained("gagan3012/keytotext-small")
['India', 'Wedding'] -> We are celebrating today in New Delhi with three wedding anniversary parties.
Select AutoNLP in the “Train” menu to fine-tune this model automatically.
- Downloads last month