Update abstractive_model.py

abstractive_model.py

@@ -1,8 +1,12 @@
 from transformers import pipeline
 
-summarizer = pipeline("summarization", model="EE21/BART-ToSSimplify")
+# Load the BART tokenizer and model
+tokenizer = AutoTokenizer.from_pretrained("EE21/BART-ToSSimplify")
+model = AutoModelForSeq2SeqLM.from_pretrained("EE21/BART-ToSSimplify")
 
+# Define the abstractive summarization function
 def summarize_with_bart(input_text):
-    # Use the pipeline to generate a summary
-    summary = summarizer(input_text)
-    return summary[0]['summary_text']
+    inputs = tokenizer.encode("summarize: " + input_text, return_tensors="pt", max_length=1024, truncation=True)
+    summary_ids = model.generate(inputs, max_length=150, min_length=40, length_penalty=2.0, num_beams=4, early_stopping=True)
+    summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+    return summary