Tensorflow 2 support

The size of the package is exceeding when having TensorFlow 2 as a requirement. Please help out!