ImportError: No module named cnn_model

[redaaa99]

Ubuntu 18.04.3 LTS
Python 3.6.9

Command:
python3 predict.py --model "./models/pretrained_model.pth" --text "I love pizza !" --max_length 150

Output:
Traceback (most recent call last):
File "predict.py", line 47, in
prediction = predict(args)
File "predict.py", line 14, in predict
state = torch.load(args.model)
File "/home/reda/.local/lib/python3.6/site-packages/torch/serialization.py", line 426, in load
return _load(f, map_location, pickle_module, **pickle_load_args)
File "/home/reda/.local/lib/python3.6/site-packages/torch/serialization.py", line 613, in _load
result = unpickler.load()
ModuleNotFoundError: No module named 'src.cnn_model'