Its probably OK to ride with a few balls short esp if you are getting a new headset.
Note that changing the headset is about the most difficult thing to do on the bike. You can do it at home with some home-made tools but bikeshops have a headset press to make sure it is well seated and aligned. For many of us home mechanics, it is the one thing we leave to the bike shop.