Update README.md

README.md

@@ -1,3 +1,40 @@
 ---
 license: mit
+language:
+- en
+tags:
+- NoSleep
+- Reddit
+- Story
+- Horror
+widget:
+- text: "[WP] "
+  example_title: "[WP] "
+datasets:
+- chloeliu/reddit_nosleep_posts
 ---
+
+# "NoSleep" Writing Prompt Generator
+Finetuned version of [GPT2](https://huggingface.co/gpt2) to facilitate generation of Writing Prompts for the [GPT-NoSleep-355m model](https://huggingface.co/DarwinAnim8or/GPT-NoSleep-355m)
+
+You can use the space linked on the right to use this model, then use the NoSleep model in tandem to generate stories!
+
+# Training Procedure
+This was trained on the 'reddt-nosleep-posts' dataset, using the "HappyTransformers" library on Google Colab.
+This model was trained for X epochs with learning rate 1e-2.
+
+# Biases & Limitations
+This likely contains the same biases and limitations as the original GPT2 that it is based on, and additionally heavy biases from the dataset.
+It likely will generate offensive output. 
+
+# Intended Use
+This model is meant for fun, nothing else.
+
+# Sample Use
+```python
+from happytransformer import GENSettings
+args_top_k = GENSettings(no_repeat_ngram_size=1, do_sample=True, top_k=80, temperature=0.4, max_length=25, early_stopping=True)
+
+result = happy_gen.generate_text("[WP] \"", args=args_top_k)
+print(result.text)
+```