File size: 265 Bytes
6a1b091
39dbdf0
 
6a1b091
 
39dbdf0
 
 
 
 
6a1b091
1
2
3
4
5
6
7
8
9
10
11
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