Update README.md

README.md

@@ -4,4 +4,27 @@ datasets:
 - CarperAI/openai_summarize_tldr
 metrics:
 -rouge1: 0.3156
----
+---
+<details> <summary><strong>Example usage</strong></summary>
+```
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("KookyGhost/GPT2-small-summarization")
+model = AutoModelForCausalLM.from_pretrained("KookyGhost/GPT2-small-summarization")
+
+prompt = "Summarize this: Reddit user shares a long story about learning to code with free online resources and eventually landing their first developer job."
+
+inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+outputs = model.generate(
+    **inputs,
+    max_new_tokens=60,
+    do_sample=True,
+    top_k=50,
+    top_p=0.95,
+    temperature=0.7
+)
+
+summary = tokenizer.decode(outputs[0], skip_special_tokens=True)
+print(summary)
+```
+</details>