walter white model in tf 2