czearing commited on
Commit
82c02b7
1 Parent(s): f622a48

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -0
README.md CHANGED
@@ -1,3 +1,17 @@
1
  ---
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
  ---
4
+
5
+ ## Story to Title
6
+ 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.
7
+
8
+ ## Usage
9
+ Example code:
10
+ ```python
11
+ from transformers import AutoTokenizer, AutoModel
12
+ tokenizer = AutoTokenizer.from_pretrained("czearing/story-to-title")
13
+ model = AutoModel.from_pretrained("czearing/czearing/story-to-title")
14
+ ```
15
+
16
+ ## License
17
+ MIT