runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/home/user/app/app.py", line 9, in <module> generator_A2B = Generator().to(device) NameError: name 'Generator' is not defined. Did you mean: 'GeneratorExit'?

Container logs: