Update README.md

README.md

@@ -8,3 +8,43 @@ tags: []
 <!-- Provide a quick summary of what the model is/does. -->
 
 Midwest indie band 24thankyou lyrics generator, finetuned from gpt2.
+
+
+## How to use:
+
+
+You can use the following code to generate lyrics
+
+```python
+
+from transformers import GPT2LMHeadModel, GPT2Tokenizer, pipeline
+
+# load model
+model_name = "smgriffin/24thankyou-lyrics-generator"
+model = GPT2LMHeadModel.from_pretrained(model_name)
+tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+
+# create text generation pipeline
+text_generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
+
+# prompt for lyrics
+prompt = f"Lyrics:"
+
+# generate and print
+generated_texts = text_generator(
+    prompt, 
+    max_length=150,
+    num_return_sequences=1,  
+    temperature=0.9,  # less than .9 results in a lot of repeated lyrics
+    top_k=50,
+    top_p=0.95,
+    do_sample=True, 
+)
+
+print("Generated Lyrics:")
+print(generated_texts[0]["generated_text"])
+
+
+```
+
+