Update README.md

README.md

@@ -16,8 +16,13 @@ inference: false
 
 > ⚠️ warning - this is a work in progress ⚠️
 
+<a href="https://colab.research.google.com/gist/pszemraj/ea0ac20dae4ad84bea4ea64543f84a85/long-t5-tglobal-xl-qmsum-wip.ipynb">
+  <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
+</a>
+
+This model is a fine-tuned version of [google/long-t5-tglobal-xl](https://huggingface.co/google/long-t5-tglobal-xl) on the `pszemraj/qmsum-cleaned` dataset. Refer to the [dataset card](https://huggingface.co/datasets/pszemraj/qmsum-cleaned) for details but this model was trained **with the task/prompt prefixes at the start of `input`** which means that **inference should be run in a similar fashion**.
+- an example of this is in the Colab notebook linked above.
 
-This model is a fine-tuned version of [google/long-t5-tglobal-xl](https://huggingface.co/google/long-t5-tglobal-xl) on the `pszemraj/qmsum-cleaned` dataset.
 It achieves the following results on the evaluation set:
 - Loss: 2.0505
 - Rouge1: 35.3881