changed the classifier model file to state_dict only file, to address pickle issues 81dfa2b verified abra cadabra commited on Feb 5