Remove tf2_py dependency for Android