model_selector.py

from transformers import AutoModelForSequenceClassification

def get_model_list():
    # List of popular pre-trained models
    models = [
        "bert-base-uncased",
        "distilbert-base-uncased",
        "roberta-base",
        "gpt2"
    ]
    return models