Update README.md

README.md

@@ -91,6 +91,7 @@ def compute_metrics(eval_preds):
     probs, labels = eval_preds
     preds = np.argmax(probs, axis=-1)
     result = {"acc": accuracy_score(labels, preds), "mcc": matthews_corrcoef(labels, preds)}
+    return result
 
 def preprocess_logits_for_metrics(logits, labels):
     return torch.softmax(logits, dim=-1)