The statement importing matplotlib.pyplot has been moved inside show_image() method.
The statement importing matplotlib.pyplot has been moved inside show_image() method.

aakhundov/tf-attend-infer-repeat