Update README.md

README.md

@@ -9,15 +9,20 @@ tags:
 - non-commercial
 - dialogue
 - chatbot
+- ai-msgbot
 
 inference: false
 ---
 
 # pszemraj/opt-peter-2.7B
 
+ <a href="https://colab.research.google.com/gist/pszemraj/26a69775c9d012051396ab5ae980f5c1/example-text-gen-pszemraj-opt-peter-2-7b.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 [facebook/opt-2.7b](https://huggingface.co/facebook/opt-2.7b) on about 80k whatsapp/text messages (mine). Please use responsibly :)
 
-Test it out on Google Colab [here](https://colab.research.google.com/gist/pszemraj/26a69775c9d012051396ab5ae980f5c1/example-text-gen-pszemraj-opt-peter-2-7b.ipynb)!
+Test it out on Google Colab by clicking the button above.
 
 ![chatdemo](https://i.imgur.com/1EgQYat.png)