error when trying to obtain predictions

[MalgiVonKokos]

Hi!

I have little experience with BERT and transformers and can't figure out how to use the model the make predictions once the additional classification layer it's trained and reloaded, I keep getting dimension errors that seem to suggest there is something wrong with input. Is it possible to use SentimentClassifier class to make predictions by prediction = model(sequence, attention_mask)? Could you possibly provide an example of applying a finetuned model that would work with your implementation? Thanks!

Malgi