runtime error

ite-packages (from torch) (2.3.1) Requirement already satisfied: nvidia-cuda-nvrtc-cu12==12.1.105 in /usr/local/lib/python3.10/site-packages (from torch) (12.1.105) Requirement already satisfied: nvidia-nccl-cu12==2.20.5 in /usr/local/lib/python3.10/site-packages (from torch) (2.20.5) Requirement already satisfied: nvidia-cusolver-cu12==11.4.5.107 in /usr/local/lib/python3.10/site-packages (from torch) (11.4.5.107) Requirement already satisfied: sympy in /usr/local/lib/python3.10/site-packages (from torch) (1.12.1) Requirement already satisfied: nvidia-cublas-cu12==12.1.3.1 in /usr/local/lib/python3.10/site-packages (from torch) (12.1.3.1) Requirement already satisfied: nvidia-curand-cu12==10.3.2.106 in /usr/local/lib/python3.10/site-packages (from torch) (10.3.2.106) Requirement already satisfied: filelock in /usr/local/lib/python3.10/site-packages (from torch) (3.15.4) Requirement already satisfied: nvidia-cuda-runtime-cu12==12.1.105 in /usr/local/lib/python3.10/site-packages (from torch) (12.1.105) Requirement already satisfied: nvidia-nvjitlink-cu12 in /usr/local/lib/python3.10/site-packages (from nvidia-cusolver-cu12==11.4.5.107->torch) (12.5.82) Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in /usr/local/lib/python3.10/site-packages (from torchvision) (10.4.0) Requirement already satisfied: numpy in /usr/local/lib/python3.10/site-packages (from torchvision) (1.26.4) Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/site-packages (from jinja2->torch) (2.1.5) Requirement already satisfied: mpmath<1.4.0,>=1.1.0 in /usr/local/lib/python3.10/site-packages (from sympy->torch) (1.3.0) Installing collected packages: torchvision, torchaudio Successfully installed torchaudio-2.3.1+cu118 torchvision-0.18.1+cu118 [notice] A new release of pip available: 22.3.1 -> 24.1.1 [notice] To update, run: pip install --upgrade pip Traceback (most recent call last): File "/home/user/app/app.py", line 9, in <module> from main import create_original ModuleNotFoundError: No module named 'main'

Container logs: