aspis commited on
Commit
2790989
1 Parent(s): 4d2a753

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -8
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 by manipulating the BookCorpus dataset by getting the different genres per book and dividing the text into paragraphs.
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