Running prediction.py

[BSharmi]

Hi,

Is there an example on running prediction.py with the model under output/models?
I tried running it via the command
python prediction.py --input_sequence='MKNLISFGVKPWWAARWETVEPEPEEPVYT' --model_path="/home/openprotein/output/models/2019-01-30_00_38_46-TRAIN-LR0_01-MB1.model"
but get error that ModuleNotFoundError: No module named 'models' while executing result = unpickler.load() in the script /home/ubuntu/.local/share/virtualenvs/openprotein-_438uSzB/lib/python3.7/site-packages/torch/serialization.py

Would appreciate any pointers.

Thank you!