Normally, I would recommend Berkeley for its outstanding history. However, according to your needs, you don’t really seem to care as much about the discipline itself, as caring about getting into law school. At least you’re honest.
And in the case of law school and your grades, I would recommend UCLA. If you’re really intent on law school and your requisite undergrad performance, the surroundings aren’t going to matter a whole lot. UCLA has quite a nice law library to study in, btw. Truthfully, it is likely you will achieve a higher GPA, presuming you are diligent. Though do be aware of quarters, vs. semesters at Berkeley (this is still the case, correct?)
That said, if you feel the context and surroundings of Berkeley will truly influence your respective performance, choose Berkeley and don’t look back.