Downloads · 30 days
46
3% of all-time downloads
keytotext-small
gagan3012/keytotext-small
keytotext-small 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
1.7K
Public
Repo size
726 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin242 MB · 99%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- t5
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Mar 11, 2021
- SHA
- bf3e0c8f
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