A Surrogate model for High Temperature Superconducting Magnets to Predict Current Distribution with Neural Network