Update README.md

README.md

@@ -1,8 +1,8 @@
 ---
-title: Class Saliency
-emoji: 😻
-colorFrom: indigo
-colorTo: pink
+title: Class Saliency Map
+emoji: 🎻
+colorFrom: red
+colorTo: purple
 sdk: gradio
 sdk_version: 2.9.4
 app_file: app.py
@@ -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 saliency 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.).