Views
No views yet
1
2from transformers import AutoModelForSequenceClassification, AutoFeatureExtractor
3
4fe = AutoFeatureExtractor.from_pretrained("DBD-research-group/BirdMAE-XCL", trust_remote_code=True) # "DBD-research-group/Bird-MAE-Base" also works
5
6model = AutoModelForSequenceClassification.from_pretrained("DBD-research-group/BirdMAE-XCL", trust_remote_code=True, num_labels=9736)
7
8print(model.config.id2label)
9
10# {0: 'ostric2',
11# 1: 'grerhe1',
12# 2: 'lesrhe2',
13# 3: 'sobkiw1',
14# ...
15# }
16