Downloads · 30 days
64
2% of all-time downloads
keytotext
gagan3012/keytotext
keytotext is a text generation model from gagan3012. Use it when you need the model to write or continue text. It is set up for transformers.
Idea is to build a model which will take keywords as inputs and generate sentences as outputs.
All-time downloads
3.2K
Public
Repo size
1.8 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin892 MB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- t5
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Mar 11, 2021
- SHA
- e72f603e
Try a prompt
Model card
From the Hugging Face model README
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.
- Task
- Text Generation
- Library
- transformers
- Type
- t5
- Created
- Mar 2, 2022
- Updated
- Mar 11, 2021