ereverter/cnn_dailymail_extractive
Viewer • Updated • 312k • 69 • 6
How to use ereverter/bert-finetuned-cnn_dailymail with Transformers:
# Load model directly
from transformers import AutoTokenizer, BertSummarizer
tokenizer = AutoTokenizer.from_pretrained("ereverter/bert-finetuned-cnn_dailymail")
model = BertSummarizer.from_pretrained("ereverter/bert-finetuned-cnn_dailymail", device_map="auto")