Story to Title
The model is based on the T5 language model and trained using a large collection of movie descriptions and corresponding titles. When given a story it will generate a corresponding title.
Usage
Example code:
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("czearing/story-to-title")
model = AutoModel.from_pretrained("czearing/story-to-title")
License
MIT
- Downloads last month
- 41