Great! Your solution is working well.
I have one more thing before this thread can be locked: ‘I got an answer which is tensor([3]). How can I convert this to a string with the classification of the fourth index?’
Thank you for your help.