How to update the model card with the best model load after a train

Hey, the best model is automatically pushed to hub according to this answer: Clarification on push_to_hub, best model, and model card

But the issue is that the model card shows the wrong metrics. Were you able to solve this issue? Can we write some code to edit the model card with the best metrics instead of the metrics of the last epoch?