Issues trying to add a third class

[Neel49]

Hello I'm having issues trying to add a third class. Specifically a "Covid" class using data from the aforementioned covid repositories in your read.me. Currently the CXR_dataset.py, data_splitter.py, and eval_finetune.py, are specifically tuned to only the "Normal" and "Tuberculosis" classes. When i tried to add a third "Covid" class by editing the code in those specific files to add an extra label in the CXR_dataset.py. I keep getting errors during my fold 1 training, surprisingly I managed to get the pretrained however.