Python Image Classification Code For Electrical Contractors