How to predict a synergy score of a certain drug combination (drugA and drugB)?

As a greenhand, I have trained the model and saved the model state_dict, but I don't know how to do a prediction with the saved model state_dict. Anyone can help me? It's better to offer me some demo codes. Thanks!