From 1d1d5bec1b30fc4fa437cba5f1bcc4bf09cc1f4c Mon Sep 17 00:00:00 2001
From: Pranav Vadrevu <pranav.vadrevu363@gmail.com>
Date: Tue, 11 Aug 2020 16:36:15 +0200
Subject: [PATCH] Create Model Card File (#6357)

---
 .../gpt2-genre-story-generator/README.md      | 32 +++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 model_cards/pranavpsv/gpt2-genre-story-generator/README.md

diff --git a/model_cards/pranavpsv/gpt2-genre-story-generator/README.md b/model_cards/pranavpsv/gpt2-genre-story-generator/README.md
new file mode 100644
index 000000000..5900081de
--- /dev/null
+++ b/model_cards/pranavpsv/gpt2-genre-story-generator/README.md
@@ -0,0 +1,32 @@
+
+# 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.