The good news, young parents, is you have a couple years to get it together. A infant wants to see your smiling face, but also won’t be scarred for life if you sometimes look away at a screen. Commit yourself to setting a good example, work on it, and know that while you will slip, you’ve also got time to improve.
Some good steps are zero phones at family eating times, kid book time, and kid bed time. I personally also think it’s important you are visibly alone when you are on a screen, in other words don't let your kids see you ignoring another adult for the screen. Like a modern smoke break.