code: def compute_metrics(eval_pred):
      logits, labels = eval_pred
      predictions = np.argmax(logits, axis=-1)
      return metric.compute(predictions=predictions, references=labels)
  
imports: import evaluate
  import numpy
  from aim.hugging_face import AimCallback
  from datasets import load_dataset
  from transformers import AutoModelForSequenceClassification, AutoTokenizer, Trainer, TrainingArguments
label_desc: Use image node_id 2 for calling a global function with example usage: compute_metrics(eval_pred) and returns: metric