README.md

---
title: Class Saliency Map
emoji: 🎻
colorFrom: red
colorTo: purple
sdk: gradio
sdk_version: 2.9.4
app_file: app.py
pinned: false
license: apache-2.0
---

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.).