Flex-Attention Mask Visualization

A Streamlit app to visualize the attention mask of the Flex-Attention model.

How to run it on your own machine

Install the requirements
```
$ pip install -r requirements.txt
```
Run the app
```
$ streamlit run streamlit_app.py
```