Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
1d1d5bec
Unverified
Commit
1d1d5bec
authored
Aug 11, 2020
by
Pranav Vadrevu
Committed by
GitHub
Aug 11, 2020
Browse files
Create Model Card File (#6357)
parent
00ce881c
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
32 additions
and
0 deletions
+32
-0
model_cards/pranavpsv/gpt2-genre-story-generator/README.md
model_cards/pranavpsv/gpt2-genre-story-generator/README.md
+32
-0
No files found.
model_cards/pranavpsv/gpt2-genre-story-generator/README.md
0 → 100644
View file @
1d1d5bec
# GPT2 Genre Based Story Generator
## Model description
GPT2 fine-tuned on genre-based story generation.
## Intended uses
Used to generate stories based on user inputted genre and starting prompts.
## How to use
#### Supported Genres
superhero, action, drama, horror, thriller, sci_fi
#### Input text format
\<
BOS>
\<
genre> Some optional text...
**Example**
:
\<
BOS>
\<
sci_fi> After discovering time travel,
```
python
# Example of usage
from
transformers
import
pipeline
story_gen
=
pipeline
(
"text-generation"
,
"pranavpsv/gpt2-genre-story-generator"
)
print
(
story_gen
(
"<BOS> <superhero> Batman"
))
```
## Training data
Initialized with pre-trained weights of "gpt2" checkpoint. Fine-tuned the model on stories of various genres.
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment