A lightweight, exhibition-ready project to showcase Vision-Transformer (VIT) AI features with a friendly UI. In particular the streamlit page can be used to vizualize the attention for each layer and head and integrated gradients of the image for the classification VIT as well as for each word of the image-to-text VIT.
Run by calling
streamlit run vit_demo.py
