Rosie Gallegos-Main apologizes 'for my poor judgment' after comparison to vaccine regulations; she'll remain at Tokyo games but must undergo diversity and inclusion training