Finding Cell selection confidence

[Dhimant404]

Hello! Does the model output a confidence score for cell selection/output confidence? If so could you please point out the code that does that so it is possible to filter the output more precisely for custom cases. I have looked at the code and couldn't find the section that had cell selection/output confidence. I'm using tapas-base-finetuned-wtq right now and want to find the same, cell selection confidence/output confidence. Is there any part of the coder I could look at/ modify to get the confidence of the prediction? I'd appreciate any help, thanks in advance.