runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/home/user/app/app.py", line 10, in <module> class CustomBlock(nn.Module): NameError: name 'nn' is not defined

Container logs: