fix: title argument

app.py

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