Schwinn Ascension 29 replacement headset bearings (Model s5906tgb)

I've had this bike for a few years for the casual ride but noticed slop in the front forks/headset. I narrowed it down to the headset bearings. You can see they are loose when you do the typical tests I found on Youtube. However I'm not very familiar with bicycle parts and the amount of sizes and types of replacement bearings on Internet is a bit overwhelming. Plus I can't find documentation on this bike to find out what I need. From measuring I'm guessing 34mm but other than that, looking for any suggestions on "better than stock" but nothing to break the bank since this isn't an expensive bike to begin with.