AI-generated Images of wavefunctions