runtime error

Traceback (most recent call last): File "/home/user/app/app.py", line 12, in <module> model = 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 340, 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 446, 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 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: 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: