Как конвертировать Tensorflow 1 в Tensorflow 2?

Всем привет знатоки, вообщем решил мигрировать на вторую версию, как я понимаю там используются функции, но как по мне документации в стиле просто замените сессии на функции, не то что нужно чтобы мигрировать полностью(не используя фрагменты с первой версии).

Не могли бы вы показать как мигрировать на примере ниже.

@worker_process_init.connect()
def init_worker(**kwargs):
    global sess, r
    with tf.io.gfile.GFile(MODEL_FILENAME, 'rb') as f:
        graph_def = tf.compat.v1.GraphDef()
        graph_def.ParseFromString(f.read())
        tf.import_graph_def(graph_def)
        
    gpu_options = tf.compat.v1.GPUOptions(per_process_gpu_memory_fraction=0.2, allow_growth=True)
    sess = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(gpu_options=gpu_options))

@app.task()
def create_mask(path):
    img_rgba = sess.run(
        sess.graph.get_tensor_by_name('import/outputs/hair:0'),
        feed_dict={'import/inputs/images:0': [image]})