Experiencing Loss - Short term disability

Did anyone take short term disability during their loss? I'm having a very hard time dealing with everything leading up to making a decision. My company is very supportive and will allow me to use it but I'm not sure taking 4-6 weeks off is way too much. I have a bunch of check up Ultrasound appointments and each time we go it gets worse. I can't even go back to work afterwards. Anytime someone asks how I am I just start crying. I don't want to breakdown at work. Thoughts or advice?