Solved by bolt25 in post #7 
              
              
                Change the following line:- 

To:- 
image_name = os.path.join(self.root_dir, str(self.labels.iloc[index, 0])) 
and your label would be- 
labels = self.labels.iloc[index, 1] 
I hope this works out for you! 
Feel free to text back! 
Regards