Treating water in a baby's lungs depends on the severity of the situation and individual circumstances. The medical team will focus on clearing the baby's airway and providing respiratory support. They might perform suctioning to remove excess water. If the baby cannot breathe properly, they might also use oxygen therapy. It is important to remember to seek medical help in a situation like this and not try home remedies as they may further the problem.