Model prediction (single predict) always returns "not yes and"

[idanvidra]

Hello @wise-east ,

I want to deploy this projects model so that I can classify stories as "yes and" or "non yes and".

I've deployed the project locally to my environment and seem to be getting "not yes and" for any pair of (prompt, response) I feed the model.

I'm using the pytorch model "bert_postacl_with_selfyesand_v2" as given here:
https://drive.google.com/file/d/1My_8SGJHR43bLHepYAl6O8WG1fbH9_9P/view

Here is an example for a sanity check I've made:
The prompt and response were taken from the dataset the model trained on
(data point id: 101430).

Any suggestions on how to get the model to predict accurately ?
Thank you for your support :)