SumBot / main.py
Tminus1's picture
Create main.py
a7494da
raw
history blame
349 Bytes
from transformers import pipeline
summarizer = pipeline("summarization")
def get_summary(text, max_length=200, min_length=50):
summary = summarizer(text, max_length, min_length, do_sample=Flase)
return summary
for i in range(10):
my_input = input("Enter the text to summerize: ")
output = get_summary(my_input)
print(my_output)