how to use the heatmap.py？

[rustalter]

I simply add it near ‘attn_x’ and set 'query_image' the picture I predict, but how to set the 'support_image' ？