runtime error

Exit code: 1. Reason: 2025-06-09 09:42:10.615877: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/lib/python3.10/site-packages/cv2/../../lib64: 2025-06-09 09:42:10.615916: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine. best_unet_model.h5: 0%| | 0.00/373M [00:00<?, ?B/s] best_unet_model.h5: 10%|▉ | 37.2M/373M [00:01<00:10, 32.5MB/s] best_unet_model.h5: 100%|█████████▉| 373M/373M [00:01<00:00, 207MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 16, in <module> model = load_model(model_path, compile=False) File "/usr/local/lib/python3.10/site-packages/keras/utils/traceback_utils.py", line 67, in error_handler raise e.with_traceback(filtered_tb) from None File "/usr/local/lib/python3.10/site-packages/keras/engine/input_layer.py", line 136, in __init__ raise ValueError(f'Unrecognized keyword arguments: {list(kwargs.keys())}') ValueError: Unrecognized keyword arguments: ['batch_shape']

Container logs: