Does medication help post partum?

I was on anxiety medication before I got pregnant and it helped me 100%. I went off it when I got pregnant. I’m so worried I’ll have anxiety/ depression after I have the baby. I fully plan to go back on medication after I have the baby. Did medication help you post partum?