Hugging Face's logo

Spaces:
probing-vits
/
class-attention-map
Runtime error

App Files Community
sayakpaul HF Staff commited on
Commit
12e48bb
·
1 Parent(s): 0dd57bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,10 +31,10 @@ def show_plot(image):
31
  _, _, ca_atn_score_dict = _MODEL.predict(preprocessed_image)
32
 
33
  result_first_block = utils.get_cls_attention_map(
34
- image, ca_atn_score_dict, block_key="ca_ffn_block_0_att"
35
  )
36
  result_second_block = utils.get_cls_attention_map(
37
- image, ca_atn_score_dict, block_key="ca_ffn_block_1_att"
38
  )
39
  return Image.fromarray(result_first_block), Image.fromarray(
40
  result_second_block
 
31
  _, _, ca_atn_score_dict = _MODEL.predict(preprocessed_image)
32
 
33
  result_first_block = utils.get_cls_attention_map(
34
+ preprocessed_image, ca_atn_score_dict, block_key="ca_ffn_block_0_att"
35
  )
36
  result_second_block = utils.get_cls_attention_map(
37
+ preprocessed_image, ca_atn_score_dict, block_key="ca_ffn_block_1_att"
38
  )
39
  return Image.fromarray(result_first_block), Image.fromarray(
40
  result_second_block