Does code is not compatible with tensorflow latest version? Which version do you use to make it work?