updated tf2 dependancy