from torchmetrics.text.bert import BERTScore
preds = ["hello there", "general kenobi"]
target = ["hello there", "master kenobi"]
metric = BERTScore()
metric.update(preds, target)
fig_, ax_ = metric.plot()