Spaces:

all-things-vits
/

class-attention-map

Runtime error

sayakpaul HF Staff commited on Jun 11, 2023

Commit

893fce9

1 Parent(s): 0e614b3

use savefig.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -79,7 +79,7 @@ def serialize_images(processed_map):
     for i in range(processed_map.shape[0]):
         plt.imshow(processed_map[i].numpy())
         plt.title(f"Attention head: {i}", fontsize=14)
-        plt.imsave(fname="attention_map_{i}.png")
 def generate_class_attn_map(image, block_id=0):

     for i in range(processed_map.shape[0]):
         plt.imshow(processed_map[i].numpy())
         plt.title(f"Attention head: {i}", fontsize=14)
+        plt.savefig(fname="attention_map_{i}.png")
 def generate_class_attn_map(image, block_id=0):