Upload README.md

README.md

@@ -1,6 +1,6 @@
 ---
 title: Class Attention Map
-emoji: 📚
+emoji: 🎻
 colorFrom: red
 colorTo: purple
 sdk: gradio
@@ -10,4 +10,11 @@ pinned: false
 license: apache-2.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
+This Space shows the spatial-class relationship of a CaiT model by plotting its
+class attention map w.r.t an input image. For speed, this Space uses the smallest
+CaiT model (`cait_xxs24_224`) but you can try it for others too (model collection
+if [here](https://tfhub.dev/sayakpaul/collections/cait/1)). You'd just need to
+adjust the code accordingly.
+
+CaiT was proposed in [Going deeper with Image Transformers](https://arxiv.org/abs/2103.17239)
+(Touvron et al.).