Spaces:

probing-vits
/

class-saliency

Runtime error

sayakpaul HF staff commited on May 2, 2022

Commit

1d20657

•

1 Parent(s): 3e32c41

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ def show_plot(image):
     # Compute the saliency map and superimpose.
     result_first_block = utils.get_cls_attention_map(
-        image, ca_atn_score_dict, block_key="ca_ffn_block_0_att"
     )
     heatmap = cv2.applyColorMap(
         np.uint8(255 * result_first_block), cv2.COLORMAP_CIVIDIS

     # Compute the saliency map and superimpose.
     result_first_block = utils.get_cls_attention_map(
+        preprocessed_image, ca_atn_score_dict, block_key="ca_ffn_block_0_att"
     )
     heatmap = cv2.applyColorMap(
         np.uint8(255 * result_first_block), cv2.COLORMAP_CIVIDIS