config.py

from metrics import BERT_FRIENDLY_TO_MODEL

# Available evaluation metrics for selection
METRIC_CHOICES = ["BLEU", "BLEURT", "ROUGE", "BERTSCORE"]
# Default metrics pre-selected in the UI
DEFAULT_METRICS = ["BLEU"]

# Available BERT models for BERTScore
BERT_CHOICES = list(BERT_FRIENDLY_TO_MODEL.keys())
# Default BERT model pre-selected in the UI
DEFAULT_BERTS = [BERT_CHOICES[0]]