2024-10-10 14:51:48 +00:00
|
|
|
from transformers import pipeline
|
|
|
|
|
|
|
|
|
|
def process(input_text):
|
2024-10-10 15:03:15 +00:00
|
|
|
summarizer = pipeline(
|
|
|
|
|
"summarization",
|
|
|
|
|
model="marianna13/flan-t5-base-summarization",
|
|
|
|
|
max_length=200,
|
|
|
|
|
min_length=10,
|
|
|
|
|
device="mps"
|
|
|
|
|
)
|
2024-10-10 14:51:48 +00:00
|
|
|
output = summarizer(input_text)[0]['summary_text']
|
|
|
|
|
return output
|
