change the model

app.py

@@ -1,7 +1,7 @@
 import torch
-from transformers import PegasusForConditionalGeneration, PegasusTokenizer
+from transformers import BertForMaskedLM
 
-model_name = 'tuner007/pegasus_paraphrase'
+model = BertForMaskedLM.from_pretrained("remi/bertabs-finetuned-extractive-abstractive-summarization")
 torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
 tokenizer = PegasusTokenizer.from_pretrained(model_name)
 model = PegasusForConditionalGeneration.from_pretrained(model_name).to(torch_device)