keytotext

Idea is to build a model which will take keywords as inputs and generate sentences as outputs.

Model:

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

Usage:

from transformers import AutoTokenizer, AutoModelWithLMHead

tokenizer = AutoTokenizer.from_pretrained("gagan3012/keytotext-small")

model = AutoModelWithLMHead.from_pretrained("gagan3012/keytotext-small")

Demo:

https://share.streamlit.io/gagan3012/keytotext/app.py

Example:

['India', 'Wedding'] -> We are celebrating today in New Delhi with three wedding anniversary parties.