tensorflow==2.5
tensorflow-addons==0.15.0