How to use fine tuned model for prediction?

[Anshullllllll]

Hii @Zhihan1996 I was implementing DNABERT2 and i found that you have provided script for pre-training and fine tuning so using your fine tuning script i had fine tuned a model for a specific type of dataset but i want to test this fine tuned on a testing dataset and predict labels just like DNABERT but there is no prediction script present.
Please let me how can i done prediction part??