Need Help with AI/Machine Learning Project

If someone could help me with my code, that would be great. It’s a project for work. Trying to train an AI to recognize flux stains on a PCB. My model will not train correctly,keep getting an error on the output. Any help is greatly apricated.