Thanks Abhilash.
I tried what you suggested and came to this error.

 File "/home/garud/Documents/GCP/greetings_nlp/main.py", line 33, in visualize_data
    test_ds)
  File "/home/garud/Documents/GCP/greetings_nlp/dataloader.py", line 74, in create_vocalb
    self.LABEL.build_vocab((train_ds, test_ds))
  File "/home/garud/Documents/GCP/greetings_nlp/venv/lib/python3.7/site-packages/torchtext/data/field.py", line 306, in build_vocab
    self.vocab = self.vocab_cls(counter, specials=specials, **kwargs)
  File "/home/garud/Documents/GCP/greetings_nlp/venv/lib/python3.7/site-packages/torchtext/vocab.py", line 75, in __init__
    words_and_frequencies = sorted(counter.items(), key=lambda tup: tup[0])
TypeError: '<' not supported between instances of 'RSICSDataset' and 'RSICSDataset'