Hugging Face's logo

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

App Files Community
2
ariG23498 commited on
Commit
9d17ea5
•
1 Parent(s): e029f9c

chore: reading image outside with context

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -45,7 +45,6 @@ with st.spinner('Processing the heat maps. Wait for it 🔥...'):
45
 
46
  utils.plot(attentions=attentions, image=preprocessed_img_orig)
47
 
48
- # Show the attention maps
49
- image = Image.open("heat_map.png")
50
-
51
  st.image(image, caption="Attention Heat Maps")
 
45
 
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")