Update README.md

README.md

@@ -22,6 +22,8 @@ co2_eq_emissions:
 # GPT-NoSleep-355m
 A finetuned version of [GPT2-Medium](https://huggingface.co/gpt2-medium) on the 'reddit-nosleep-posts' dataset. (Linked above)
 
+**TIP** You can find a larger, more capable version of the model here: [GPT-NoSleep-1.5b](https://huggingface.co/DarwinAnim8or/GPT-NoSleep-1.5b)
+
 # 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.
@@ -31,4 +33,21 @@ This likely contains the same biases and limitations as the original GPT2 that i
 It likely will generate offensive output. 
 
 # Intended Use
-This model is meant for fun, nothing else.
+This model is meant for fun, nothing else.
+
+# Sample code
+```python
+#Import model:
+from happytransformer import HappyGeneration
+happy_gen = HappyGeneration("GPT2", "DarwinAnim8or/GPT-NoSleep-355m")
+
+#Set generation settings:
+from happytransformer import GENSettings
+args_top_k = GENSettingsGENSettings(no_repeat_ngram_size=3, do_sample=True, top_k=80, temperature=0.8, max_length=150, early_stopping=False)
+
+#Generate a response:
+result = happy_gen.generate_text("[WP] We don't go to the forest at night [RESPONSE] ", args=args_top_k)
+
+print(result)
+print(result.text)
+```