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