Getting Issues while running the deeponet on Darcy Flow

[Karn3003]

I am trying to run the code on darcy flow and getting following error. Please look into it. Thanks

AttributeError Traceback (most recent call last)
/tmp/ipykernel_238106/3930051686.py in
1 m = 29 ** 2
2 activation = "relu"
----> 3 branch = tf.keras.Sequential([
4 tf.keras.layers.InputLayer(input_shape=(m,)),
5 tf.keras.layers.Reshape((29, 29, 1)),

AttributeError: 'function' object has no attribute 'keras'

[lululxvi]

TensorFlow may not be installed correctly.