Error al ejecutar un programa

Hola a todos y todas! Estoy probando un modelo de segmentación y estoy corriendo una prueba (propuesta por el creador del codigo) el tema es que me da el siguiente error

ImportError: keras.optimizers.legacy is not supported in Keras 3. When using tf.keras, to continue using a tf.keras.optimizers.legacy optimizer, you can install the tf_keras package (Keras 2) and set the environment variable TF_USE_LEGACY_KERAS=True to configure TensorFlow to use tf_keras when accessing tf.keras

Al correr la siguiente linea de comando:

model = seg.Unet()

model.compile(optimizer=Adam(), loss=seg.weighted_crossentropy, metrics=[“accuracy”])

model.load_weights(‘./checkpoints/seg_model’);

the SAM model checkpoints can be downloaded from: https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth

sam = sam_model_registry"default"

Si alguien me puede ayudar con el error estaré muy agradecido. Estoy trabajando con Júpiter Notebook a través de Visual Studio
Muchas gracias