story-to-title / README.md
czearing's picture
Update README.md
82c02b7
|
raw
history blame
471 Bytes
metadata
license: mit

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/czearing/story-to-title")

License

MIT