Update abstractive_summarization.py

abstractive_summarization.py

@@ -5,8 +5,8 @@ tokenizer = AutoTokenizer.from_pretrained("EE21/BART-ToSSimplify")
 model = AutoModelForSeq2SeqLM.from_pretrained("EE21/BART-ToSSimplify")
 
 # Load the BART-large-cnn tokenizer and model
-tokenizer_cnn = BartTokenizer.from_pretrained("facebook/bart-large-cnn")
-model_cnn = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn")
+tokenizer_cnn = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
+model_cnn = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
 
 # Function to summarize using the fine-tuned BART model
 def summarize_with_bart_ft(input_text):