runtime error

Exit code: 1. Reason: .gitattributes: 0%| | 0.00/1.52k [00:00<?, ?B/s] .gitattributes: 100%|██████████| 1.52k/1.52k [00:00<00:00, 13.8MB/s] README.md: 0%| | 0.00/768 [00:00<?, ?B/s] README.md: 100%|██████████| 768/768 [00:00<00:00, 9.02MB/s] model.pkl: 0%| | 0.00/196M [00:00<?, ?B/s] model.pkl: 100%|█████████▉| 196M/196M [00:00<00:00, 284MB/s] pyproject.toml: 0%| | 0.00/163 [00:00<?, ?B/s] pyproject.toml: 100%|██████████| 163/163 [00:00<00:00, 1.05MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 10, in <module> learn = from_pretrained_fastai(repo_id) File "/usr/local/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn return fn(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/huggingface_hub/fastai_utils.py", line 339, in from_pretrained_fastai return load_learner(os.path.join(storage_folder, "model.pkl")) File "/usr/local/lib/python3.10/site-packages/fastai/learner.py", line 448, in load_learner try: res = torch.load(fname, map_location=map_loc, pickle_module=pickle_module) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 1360, in load return _load( File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 1848, in _load result = unpickler.load() File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 1837, in find_class return super().find_class(mod_name, name) AttributeError: Custom classes or functions exported with your `Learner` not available in namespace. Re-declare/import before loading: Can't get attribute 'TargetMaskConvertTransform' on <module '__main__' from '/home/user/app/app.py'>

Container logs: