There isn't really a snail that will eat diatoms. But the good news is, diatoms will go away on their own. Are you using tap water? If so, the diatoms may not go away, as diatoms use silicates to build their skeletons. Silicates are found in tap water and certain types of sand/substrates.
Having a few astraea snails won't hurt though.