Hi, I have managed to train a model using trainer. I evaluated some results whilst the model was still on the disk using ‘trainer.predict()’. I then used trainer.save_model() and now want to load it up for usage again. By saving, I got three files in my drive;

pytorch_model.bin
config.json
training_args.json

I am assuming the model is pytorch_model.bin but I am unsure how do I load it up? I trained a classification model so do I call again;
model = AutoModelForSequenceClassification.from_pretrained(**filepath**, 
                                                           num_labels=5,
                                                           output_attentions = False, #
                                                           output_hidden_states = False
                                                           )

where filepath is the path to pytorch_model.bin? do I then call again a trainer and send the model to that to again use trainer.predict() for new predictions?