Thanks for detail explanation.

I guess you also need to import tourch and numpy as np in the final code.

Finally print the result may be print(preds)