A.- J._ souffre depuis son enfance d'épilepsie, affection pour laquelle elle est suivie de manière régulière par le docteur A._, neurologue. Elle a néanmoins pu se charger elle-même de la tenue de son ménage et de l'éducation de son fils durant plusieurs années, tout en travaillant, depuis le mois d'avril 1996, deux demi-journées par semaine en qualité de vendeuse dans une laiterie. En raison de douleurs dorsales, J._ résilia son contrat de travail pour le 30 novembre 1997 et fut opérée d'une hernie discale le 19 décembre de la même année. Elle déposa alors une demande de prestations à l'Office de l'assurance-invalidité pour le canton de Vaud (ci-après : l'office AI). Selon les renseignements obtenus par celui-ci auprès du médecin traitant, la doctoresse B._, rhumatologue, un reclassement professionnel était nécessaire, dans la mesure où l'ancienne activité lucrative de l'assurée comportait des tâches à effectuer dans des positions contre-indiquées (penchée en avant ou de côté) ainsi que le port de lourdes charges. Pour sa part, le docteur A._ a suggéré l'allocation d'une rente entière d'invalidité. Une enquête économique sur le ménage fut par ailleurs réalisée en novembre 1998, au terme de laquelle une incapacité de travail de 34,6 % fut retenue pour l'ensemble des tâches ménagères. Enfin, l'office AI confia une expertise pluridisciplinaire au Centre d'observation médicale de l'assurance-invalidité (ci-après : COMAI), dont le rapport fit état d'une capacité de travail résiduelle de 50 %, pouvant progressivement être augmentée jusqu'à 100 %, dans une activité ne nécessitant pas le port de charges supérieures à 10 kg ni l'exécution de mouvements répétitifs de torsion et de flexion du tronc; toujours selon ce rapport, les mouvements en porte-à-faux et les travaux statiques de durée prolongée devaient également être évités par l'assurée. Sur la base de ces renseignements, l'office AI rejeta la demande de prestations de J._, par décision du 23 novembre 2000. B.- Cette décision fut déférée par l'assurée au Tribunal des assurances du canton de Vaud, qui rejeta le recours par jugement du 7 mai 2001. C.- J._ interjette un recours de droit administratif contre ce jugement, dont elle demande l'annulation. Elle conclut, en substance, à l'allocation d'une rente d'invalidité, subsidiairement à la mise en oeuvre de mesures de réadaptation d'ordre professionnel. L'intimé conclut au rejet du recours, alors que l'Office fédéral des assurances sociales ne s'est pas déterminé. 