Fix Datamodules

[SeanNaren]

Datamodules should work with minimal code wrapping HF's dataset classes. Nate's original code defined a lot of conditions/variables that should be extracted from HF's dataset class.

Create datamodule that can take in dataset names and generate the datasets for Text Classification. Then it should be made to fit all types of data structures via inheritance if needed, and the text classification code is extracted out.

I.e TextClassificationDataModule etc IF needed (preferably everything fits one, but if it gets too big then we need so split).

[github-actions]

Hi! thanks for your contribution!, great first issue!