Update README.md

README.md

@@ -5,11 +5,17 @@ datasets:
 metrics:
 -rouge1: 0.3156
 ---
-<details> <summary><strong>Example usage</strong></summary>
-```
+
+### Example usage
+
+<p>
+
+</p>
+
+```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("KookyGhost/GPT2-small-summarization")
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
 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."
@@ -26,5 +32,3 @@ outputs = model.generate(
 
 summary = tokenizer.decode(outputs[0], skip_special_tokens=True)
 print(summary)
-```
-</details>