Hi. Thank you for the great repo. I would like to know if it would be possible to make the code compatible with TensorFlow 2?