Fixed minor bug in upload API. Features model had mismatching labels for images