Update README.md
Browse files
README.md
CHANGED
@@ -13,13 +13,13 @@ Gpt-2 for short story generation with genres.
|
|
13 |
## Model description
|
14 |
|
15 |
Gpt-2 model fine-tuned on sample of BookCorpus dataset for short story generation, allows for the following genres (tokens to use as input under parenthesis):
|
16 |
-
- Romance (<romance>)
|
17 |
-
- Adventure (<adventure>)
|
18 |
-
- Mystery & detective (<mystery-&-detective>)
|
19 |
-
- Fantasy (<fantasy>)
|
20 |
-
- Humor & comedy (humor-&-comedy)
|
21 |
-
- Paranormal (paranormal)
|
22 |
-
- Science fiction (science-fiction)
|
23 |
|
24 |
Heavily inspired by https://huggingface.co/pranavpsv
|
25 |
## Intended uses & limitations
|
@@ -46,6 +46,6 @@ This can be used for text generation.
|
|
46 |
|
47 |
## Training data
|
48 |
|
49 |
-
The model was trained
|
50 |
|
51 |
|
|
|
13 |
## Model description
|
14 |
|
15 |
Gpt-2 model fine-tuned on sample of BookCorpus dataset for short story generation, allows for the following genres (tokens to use as input under parenthesis):
|
16 |
+
- Romance ("<romance>")
|
17 |
+
- Adventure ("<adventure>")
|
18 |
+
- Mystery & detective ("<mystery-&-detective>")
|
19 |
+
- Fantasy ("<fantasy>")
|
20 |
+
- Humor & comedy ("<humor-&-comedy>")
|
21 |
+
- Paranormal ("<paranormal>")
|
22 |
+
- Science fiction ("<science-fiction>")
|
23 |
|
24 |
Heavily inspired by https://huggingface.co/pranavpsv
|
25 |
## Intended uses & limitations
|
|
|
46 |
|
47 |
## Training data
|
48 |
|
49 |
+
The model was trained using the BookCorpus dataset by getting the different genres per book and dividing the text into paragraphs.
|
50 |
|
51 |
|