Views
No views yet
bert-sci-am is a BERT-family model trained for scientific literature argument mining. At low-level it performs sequence classification. This version is trained on 3-class classification on (david-inf/am-nlp-abstrct)[david-inf/am-nlp-abstrct] forked from pie/abstrct dataset.1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2
3def load_model():
4 """Load model from hub"""
5 checkpoint = "david-inf/bert-sci-am"
6 model = AutoModelForSequenceClassification.from_pretrained(
7 checkpoint, num_labels=3)
8 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
9 return model, tokenizer
10
11model, tokenizer = load_model()