I'm struggling to get the right versions of tensorflow to run this. It seems to want python 2.7. Can you confirm that this works with tensorflow2x?