Update README.md

README.md

@@ -9,3 +9,13 @@ pinned: false
 
 ![](https://i.imgur.com/kAShjbs.gif)
 
+We study the representations learned by different families of vision transformers. 
+
+Here are some spaces that you can explore while you're here:
+
+* [Generating attention heatmaps from images](https://huggingface.co/spaces/probing-vits/attention-heat-maps)
+* [Generating attention rollout plots from images] (soon)
+
+If you'd like to dig into the code then check out [this repository](https://github.com/sayakpaul/probing-vits). If
+you're excited and want to generate attention heatmaps from videos (like the one shown above) then check out
+[this Notebook](https://github.com/sayakpaul/probing-vits/blob/main/notebooks/dino-attention-maps-video.ipynb).