The geomagnetic storm and Kp prediction using Wasserstein transformer