tensorflow2.x系でのSessionの書き換えについて

質問

マイナビbooks「実践GAN」内のコードに関する質問です。
下記コードでprogressive GANを走らせようとすると下記エラーが発生します。
tensorflow2.xからSessionが廃止されたことまでは理解したのですが
tensorflowを扱うのが初めてで、どう書き換えればエラー回避できるのかがわかりません。
どなたかお詳しい方、ご教授いただけないでしょうか？

・バージョン
python 3.6.8
tensorflow-gpu 2.3.0
tensorflow-hub 0.10.0

エラーメッセージ

AttributeError: module 'tensorflow' has no attribute 'Session'

ソースコード

python36
1import matplotlib.pyplot as plt
2import tensorflow as tf
3import tensorflow_hub as hub
4
5with tf.Graph().as_default():
6    # import the progressive GAN from TFHub
7    module = hub.Module("https://tfhub.dev/google/progan-128/1")
8    # latent dimension that gets 
9    latent_dim = 512
10
11    # Change the seed to get different faces.
12    latent_vector = tf.random.normal([1, latent_dim], seed=1337)
13
14    # Uses module to generate images from the latent space.
15    interpolated_images = module(latent_vector)
16
17    # runs the tensorflow session and gets back the image in shape (1,128,128,3)
18    with tf.Session() as session:
19        session.run(tf.global_variables_initializer())
20        image_out = session.run(interpolated_images)
21
22plt.imshow(image_out.reshape(128,128,3))
23plt.show()