diff --git a/Chirag_Dalai_AIML/text_summarizer.py b/Chirag_Dalai_AIML/text_summarizer.py new file mode 100644 index 0000000..659a12f --- /dev/null +++ b/Chirag_Dalai_AIML/text_summarizer.py @@ -0,0 +1,14 @@ +import warnings +warnings.filterwarnings("ignore") + +from transformers import pipeline + +summarizer = pipeline("summarization", model="facebook/bart-large-cnn") + +print("Enter the text you want to summarize:") +text = input() + +summary = summarizer(text, max_length=50, min_length=20, do_sample=False) + +print("\nSummary:") +print(summary[0]['summary_text'])