Transformers How to use jmt-r/predict-ai-abstract with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, MyBERTClassifier
tokenizer = AutoTokenizer.from_pretrained("jmt-r/predict-ai-abstract")
model = MyBERTClassifier.from_pretrained("jmt-r/predict-ai-abstract", device_map="auto") 
