Like Celstu1 suggested, it could be allergies and baby benedryl would help if that's the case. Have you checked her skin? Maybe the flea bath dried out her skin? I spritz my two b/c they get dry skin sometimes.. Other than that, I'd like to suggest you comb her often with a flea comb for at least couple days. Just to be sure the fleas haven't come back. Better safe then sorry.
I hope she feels better soon.