runtime error

Traceback (most recent call last): File "/home/user/app/app.py", line 4, in <module> lstm = torch.load('lstm.h5') File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 1025, in load return _load(opened_zipfile, File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 1446, in _load result = unpickler.load() File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 1439, in find_class return super().find_class(mod_name, name) AttributeError: Can't get attribute 'LSTM' on <module '__main__' from '/home/user/app/app.py'>

Container logs: