Recommended way to make inferences

[jaaxk]

Hi, I've been searching through this code and see how it's able to evaluate whole datasets, but don't see any way to actually make predictions of DDIs with the model. I've trained it and have the .pth file but don't see any methods that are designed to/that I can manipulate to be able to make a single inference. Is there a recommended way to do this? Thank you.