fix: serialization issues.

app.py

@@ -80,7 +80,7 @@ def serialize_images(processed_map):
         plt.imshow(processed_map[i].numpy())
         plt.title(f"Attention head: {i}", fontsize=14)
         plt.axis("off")
-        plt.savefig(fname="attention_map_{i}.png")
+        plt.savefig(fname=f"attention_map_{i}.png")
 
 
 def generate_class_attn_map(image, block_id=0):

gradio_cached_examples/14/log.csv

@@ -0,0 +1,2 @@
+output,flag,username,timestamp
+/Users/sayakpaul/Downloads/class-attention-map/gradio_cached_examples/14/output/24ed4fad-3279-4814-ba76-b4c411c673a0,,,2023-06-11 11:55:03.515035

gradio_cached_examples/14/output/24ed4fad-3279-4814-ba76-b4c411c673a0/captions.json

@@ -0,0 +1 @@
+{"/Users/sayakpaul/Downloads/class-attention-map/gradio_cached_examples/14/output/24ed4fad-3279-4814-ba76-b4c411c673a0/76269f58a7c390191fe41c6e016b4904749cd456/attention_map_i.png": null}