Spaces:

probing-vits
/

attention-heat-maps

Runtime error

ariG23498 HF Staff commited on Apr 11, 2022

Commit

e670f36

1 Parent(s): 3c26522

chore: adding title

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,5 +46,6 @@ with st.spinner("Generating the heat maps... HOLD ON!"):
 	utils.plot(attentions=attentions, image=preprocessed_img_orig)
 # Show the attention maps
 image = Image.open("heat_map.png")
 st.image(image, caption="Attention Heat Maps")

 	utils.plot(attentions=attentions, image=preprocessed_img_orig)
 # Show the attention maps
+st.title("Attention 🔥 Maps")
 image = Image.open("heat_map.png")
 st.image(image, caption="Attention Heat Maps")