remove line 22 . tf.compat.v1.disable_eager_execution() and miracle happens