abstractive_model.py

from transformers import pipeline

summarizer = pipeline("summarization", model="EE21/BART-ToSSimplify")

def summarize_with_bart(input_text):
    # Use the pipeline to generate a summary
    summary = summarizer(input_text)
    return summary[0]['summary_text']