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 |
|