fix colab link

README.md

@@ -282,10 +282,11 @@ model-index:
 
 # Longformer Encoder-Decoder (LED) for Narrative-Esque Long Text Summarization
 
+<a href="https://colab.research.google.com/gist/pszemraj/3eba944ddc9fc9a4a1bfb21e83b57620/summarization-token-batching.ipynb">
+  <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
+</a>
 
- [![colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/gist/pszemraj/3eba944ddc9fc9a4a1bfb21e83b57620/summarization-token-batching.ipynb)
-
-A fine-tuned version of [allenai/led-large-16384](https://huggingface.co/allenai/led-large-16384) on the BookSum dataset.
+A fine-tuned version of [allenai/led-large-16384](https://huggingface.co/allenai/led-large-16384) on the `BookSum` dataset.
 
 Goal: a model that can generalize well and is useful in summarizing long text in academic and daily usage. The result works well on lots of text and can handle 16384 tokens/batch (_if you have the GPU memory to handle that_)
 
@@ -322,17 +323,15 @@ summarizer = pipeline(
 wall_of_text = "your words here"
 
 result = summarizer(
-           wall_of_text,
-           min_length=16, 
-           max_length=256,
-           no_repeat_ngram_size=3, 
-           encoder_no_repeat_ngram_size =3,
-           repetition_penalty=3.5,
-           num_beams=4,
-           early_stopping=True,
-    )
-
-
+    wall_of_text,
+    min_length=16,
+    max_length=256,
+    no_repeat_ngram_size=3,
+    encoder_no_repeat_ngram_size=3,
+    repetition_penalty=3.5,
+    num_beams=4,
+    early_stopping=True,
+)
 ```