Update generate_attention_figure.py

[relh]

I think the dataset gets redefined a few lines later but maybe we need the first initialization for the ViewSampler? not sure..

Probably something similar in the other fig files

[relh]

Also see a key error for overfit to scene but might be because only downloaded the subset of re10k.