Hugging Face's logo

Spaces:
all-things-vits
/
class-attention-map
Runtime error

App Files Community
sayakpaul HF staff commited on
Commit
045d37d
•
1 Parent(s): deda76f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def create_attn_extractor(model, block_id=0):
29
 
30
 
31
  def get_cls_attention_map(
32
- image, attn_score_dict=out, block_key="blocks_token_only.0.attn.softmax"
33
  ):
34
  """Prepares attention maps so that they can be visualized."""
35
  w_featmap = image.shape[3] // patch_size
 
29
 
30
 
31
  def get_cls_attention_map(
32
+ image, attn_score_dict, block_key="blocks_token_only.0.attn.softmax"
33
  ):
34
  """Prepares attention maps so that they can be visualized."""
35
  w_featmap = image.shape[3] // patch_size