Using AI to better predict severe weather