Mona_Jalal:

please let me know if you might have any suggestions


Is the problem that the model isn’t able to detect this class and you would like to improve the model itself or would you rather like to fix the None error in case no valid classes were detected by the model?
I would personally try to fix the second issue first, as you might face the same error for other classes in case the model is also unable to detect them.