diff --git a/examples/seq2seq/bertabs/run_summarization.py b/examples/seq2seq/bertabs/run_summarization.py index 5dd8f22729..cf4806d221 100644 --- a/examples/seq2seq/bertabs/run_summarization.py +++ b/examples/seq2seq/bertabs/run_summarization.py @@ -30,7 +30,7 @@ Batch = namedtuple("Batch", ["document_names", "batch_size", "src", "segs", "mas def evaluate(args): tokenizer = BertTokenizer.from_pretrained("bert-base-uncased", do_lower_case=True) - model = BertAbs.from_pretrained("bertabs-finetuned-cnndm") + model = BertAbs.from_pretrained("remi/bertabs-finetuned-extractive-abstractive-summarization") model.to(args.device) model.eval()