demo/config.py

SAMPLE_SIZE = 1000
MAX_SAMPLE_SIZE = 5000
GRADIO_THEME = None

ALL_LMS = [
    # tiny
    "prajjwal1/bert-tiny", "arnir0/Tiny-LLM",
    "sentence-transformers/all-MiniLM-L12-v2", "google/electra-small-discriminator",
    "distilbert-base-cased", "typeform/distilroberta-base-v2",

    # small
    "bert-base-cased", "roberta-base", "google/electra-base-discriminator", "microsoft/deberta-v3-base",
    "KISTI-AI/scideberta", "sentence-transformers/all-mpnet-base-v2", "huggingface/CodeBERTa-small-v1",
    "FacebookAI/xlm-roberta-base", "microsoft/mdeberta-v3-base", "HuggingFaceTB/SmolLM2-135M"
]

PRESELECTED_LMS = [
    "prajjwal1/bert-tiny",
    "sentence-transformers/all-MiniLM-L12-v2",
    "arnir0/Tiny-LLM",
    "google/electra-small-discriminator",
]