Hugging Face's logo

Spaces:
probing-vits
/
attention-heat-maps
Runtime error

App Files Community
2
ariG23498 commited on
Commit
e670f36
•
1 Parent(s): 3c26522

chore: adding title

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -46,5 +46,6 @@ with st.spinner("Generating the heat maps... HOLD ON!"):
46
  utils.plot(attentions=attentions, image=preprocessed_img_orig)
47
 
48
  # Show the attention maps
 
49
  image = Image.open("heat_map.png")
50
  st.image(image, caption="Attention Heat Maps")
 
46
  utils.plot(attentions=attentions, image=preprocessed_img_orig)
47
 
48
  # Show the attention maps
49
+ st.title("Attention 🔥 Maps")
50
  image = Image.open("heat_map.png")
51
  st.image(image, caption="Attention Heat Maps")