It may be more 'harmful' for babies to see parents paying attention to screens than it is for them to watch the screen themselves.
(They also become very good at telling if you're really "looking" at the phone or just pretending to look at something.)